iicd/MasterThesis
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

for notes

Browse Source
This commit is contained in:
Mhrooz
2024-07-09 14:23:53 +02:00
parent 7a071fa658 37fdde8e83
commit ad37f7a57d
3 changed files with 1633 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
UNet/train.py
View File

@@ -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)