add some comments

.gitignore

@@ -1,8 +1,14 @@
 ./flowers/*
 .DS_Store
-./UNet/train_image/*
+UNet/train_image/*
-./UNet/params/*
+UNet/params/*
-./UNet/__pycache__/*
+UNet/__pycache__/*
+Logs_Checkpoints/
 data/
 archive.zip
 flowers/*
+VOCdevkit/
+inference_results
+.train.py.swp
+VOCdevkit_08-Jun-2007.tar
+wget-log

UNet/train.py

@@ -7,9 +7,9 @@ from net import *
 from torchvision.utils import save_image
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-weight_path = r'/Users/hanzhangma/Nextcloud/mhz/Study/SS24/MasterThesis/UNet/params/unet.pth'
+weight_path = r'/home/stud/hanzhang/MasterThesis/UNet/params/unet.pth'
-data_path = r'/Users/hanzhangma/Document/DataSet/VOC2007'
+data_path = r'/home/stud/hanzhang/MasterThesis/VOCdevkit/VOCdevkit/VOC2007'
-save_path = r'/Users/hanzhangma/Nextcloud/mhz/Study/SS24/MasterThesis/Unet/train_image'
+save_path = r'/home/stud/hanzhang/MasterThesis/UNet/train_image'
 
 if __name__ == '__main__':
     data_loader = DataLoader(MyDataset(data_path), batch_size= 4, shuffle=True)