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:20:03 +02:00
parent dabc2495a9
commit 7a071fa658
3 changed files with 13 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
UNet/train.py
View File

@@ -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'
data_path = r'/Users/hanzhangma/Document/DataSet/VOC2007'
save_path = r'/Users/hanzhangma/Nextcloud/mhz/Study/SS24/MasterThesis/Unet/train_image'
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'
if __name__ == '__main__':
data_loader = DataLoader(MyDataset(data_path), batch_size= 4, shuffle=True)