added some notes

2026-04-10 12:33:44 +00:00 · 2021-02-04 13:39:41 +01:00
parent fa6ab9161e
commit 2d876db44c
11 changed files with 191 additions and 1 deletions
--- a/ML/Pytorch/image_segmentation/semantic_segmentation_unet/pycache/dataset.cpython-38.pyc
+++ b/ML/Pytorch/image_segmentation/semantic_segmentation_unet/pycache/dataset.cpython-38.pyc
--- a/ML/Pytorch/image_segmentation/semantic_segmentation_unet/pycache/model.cpython-38.pyc
+++ b/ML/Pytorch/image_segmentation/semantic_segmentation_unet/pycache/model.cpython-38.pyc
--- a/ML/Pytorch/image_segmentation/semantic_segmentation_unet/pycache/utils.cpython-38.pyc
+++ b/ML/Pytorch/image_segmentation/semantic_segmentation_unet/pycache/utils.cpython-38.pyc
--- a/ML/Pytorch/image_segmentation/semantic_segmentation_unet/data/note.txt
+++ b/ML/Pytorch/image_segmentation/semantic_segmentation_unet/data/note.txt
@@ -0,0 +1,7 @@
+download the data from kaggle and put in folders similar to the video, specifically you should have 4
+folders:
+
+- train_images
+- train_masks
+- val_images
+- val_masks
--- a/ML/Pytorch/image_segmentation/semantic_segmentation_unet/saved_images/note.txt
+++ b/ML/Pytorch/image_segmentation/semantic_segmentation_unet/saved_images/note.txt
@@ -0,0 +1 @@
+it will put some saved preds here so you can see how it looks like
--- a/ML/Pytorch/image_segmentation/semantic_segmentation_unet/train.py
+++ b/ML/Pytorch/image_segmentation/semantic_segmentation_unet/train.py
@@ -22,7 +22,7 @@ NUM_WORKERS = 2
 IMAGE_HEIGHT = 160  # 1280 originally
 IMAGE_WIDTH = 240  # 1918 originally
 PIN_MEMORY = True
-LOAD_MODEL = True
+LOAD_MODEL = False
 TRAIN_IMG_DIR = "data/train_images/"
 TRAIN_MASK_DIR = "data/train_masks/"
 VAL_IMG_DIR = "data/val_images/"
				`@@ -0,0 +1 @@`
				`it will put some saved preds here so you can see how it looks like`