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

add some comments

Browse Source
This commit is contained in:
Hanzhang Ma 2024-04-16 14:33:51 +02:00
parent 2cad997a27
commit 37fdde8e83
3 changed files with 1566 additions and 68 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
.gitignore vendored
View File

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

6
UNet/train.py
View File

@ -7,9 +7,9 @@ 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')
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__': 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)

1616
diffusion.ipynb
View File

File diff suppressed because one or more lines are too long