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
commit ad37f7a57d
3 changed files with 1633 additions and 59 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
.gitignore vendored
View File

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

6
UNet/train.py
View File

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

1674
diffusion.ipynb
View File

File diff suppressed because one or more lines are too long