From dc63ed15bf5205b4fd4506bde43f20dd9bc88702 Mon Sep 17 00:00:00 2001 From: matixezor Date: Mon, 31 May 2021 23:16:29 +0200 Subject: [PATCH] add helpers --- .../neural_network/helpers.py | 24 +++++++++++++++++++ 1 file changed, 24 insertions(+) create mode 100644 src/machine_learning/neural_network/helpers.py diff --git a/src/machine_learning/neural_network/helpers.py b/src/machine_learning/neural_network/helpers.py new file mode 100644 index 0000000..1fc7869 --- /dev/null +++ b/src/machine_learning/neural_network/helpers.py @@ -0,0 +1,24 @@ +import os + +from torchvision.transforms import transforms + + +main_path = os.path.dirname(os.path.abspath(__file__)) + +img_main_path = os.path.dirname(main_path) +for i in range(2): + img_main_path = os.path.dirname(img_main_path) + +img_main_path = os.path.join(img_main_path, 'images/learning') + +train_path = os.path.join(img_main_path, 'training') +test_path = os.path.join(img_main_path, 'test') +prediction_path = os.path.join(img_main_path, 'prediction') + +transformer = transforms.Compose([ + transforms.Resize((150, 150)), + transforms.RandomHorizontalFlip(), + transforms.ToTensor(), + transforms.Normalize([0.5, 0.5, 0.5], + [0.5, 0.5, 0.5]) +])