for notes

.gitignore

@@ -3,8 +3,20 @@
 UNet/train_image/*
 UNet/params/*
 UNet/__pycache__/*
+<<<<<<< HEAD
 UNet/test_image
 data/
 archive.zip
 flowers/*
 UNet/result/result.jpg
+=======
+Logs_Checkpoints/
+data/
+archive.zip
+flowers/*
+VOCdevkit/
+inference_results
+.train.py.swp
+VOCdevkit_08-Jun-2007.tar
+wget-log
+>>>>>>> 37fdde8e83ce6de72d8d7226f22343e79b8a56d0

UNet/train.py

@@ -7,9 +7,15 @@ from net import *
 from torchvision.utils import save_image
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+<<<<<<< HEAD
 weight_path = r'D:\\MasterThesis\\UNet\\params\\unet.pth'
 data_path = r'D:\\MasterThesis\\data\\VOCdevkit\\VOC2007'
 save_path = r'D:\\MasterThesis\\UNet\\train_image'
+=======
+weight_path = r'/home/stud/hanzhang/MasterThesis/UNet/params/unet.pth'
+data_path = r'/home/stud/hanzhang/MasterThesis/VOCdevkit/VOCdevkit/VOC2007'
+save_path = r'/home/stud/hanzhang/MasterThesis/UNet/train_image'
+>>>>>>> 37fdde8e83ce6de72d8d7226f22343e79b8a56d0
 
 if __name__ == '__main__':
     data_loader = DataLoader(MyDataset(data_path), batch_size= 4, shuffle=True)