### This is a data training format README
Detailed reference source code is in the `get dataset` method in `utilsutils.py`.
Automatically divide labels torchvision.datasets.ImageFolder(args.dataset_path, transform=transforms)
If the dataset is as follow:
dataset_path/class_1/image_1.jpg
dataset_path/class_1/image_2.jpg
...
dataset_path/class_2/image_1.jpg
dataset_path/class_2/image_2.jpg
...
'dataset_path' is the root directory of the dataset, 'class_1', 'class_2', etc. are different categories in the dataset, and each category contains several image files.
Use the 'ImageFolder' class to conveniently load image datasets with this folder structure, and automatically assign corresponding labels to each image.
You can specify the root directory where the dataset is located by passing the 'dataset_path' parameter, and perform operations such as image preprocessing and label conversion through other optional parameters.
没有合适的资源?快使用搜索试试~ 我知道了~
IDDM(工业,景观,动画…),支持DDPM, DDIM, web和多gpu分布式训练 Pytorch实现、生成模型、扩散模型、
共84个文件
jpg:40个
py:35个
png:4个
1.该资源内容由用户上传,如若侵权请联系客服进行举报
2.虚拟产品一经售出概不退款(资源遇到问题,请及时私信上传者)
2.虚拟产品一经售出概不退款(资源遇到问题,请及时私信上传者)
版权申诉
1 下载量 24 浏览量
2024-02-22
10:22:10
上传
评论
收藏 2.34MB ZIP 举报
温馨提示
IDDM(工业,景观,动画…),支持DDPM, DDIM, web和多gpu分布式训练。Pytorch实现、生成模型、扩散模型、分布式训练
资源推荐
资源详情
资源评论
收起资源包目录
Integrated-Design-Diffusion-Model-main.zip (84个子文件)
Integrated-Design-Diffusion-Model-main
tools
__init__.py 210B
deploy.py 6KB
generate.py 7KB
train.py 18KB
assets
nrsd_248_ema.jpg 4KB
cifar_294_ema.jpg 11KB
animate_face_488_ema.jpg 8KB
neudet_240_ema.jpg 27KB
animate_face_459_ema.jpg 9KB
neudet_270_ema.jpg 5KB
IDDM_training.png 353KB
neudet_298_ema.jpg 26KB
neu160_2.jpg 5KB
nrsd_188_ema.jpg 4KB
neudet_244_ema.jpg 25KB
neudet_290_ema.jpg 5KB
neudet_276_ema.jpg 6KB
neu160_5.jpg 3KB
nrsd_285_ema.jpg 4KB
nrsd_180_ema.jpg 5KB
nrsd_218_ema.jpg 4KB
pytorch.png 7KB
Pycharm.png 10KB
nrsd_203_ema.jpg 4KB
neudet_265_ema.jpg 6KB
nrsd_210_ema.jpg 4KB
wood_495.jpg 7KB
animate_face_499_ema.jpg 8KB
neu160_3.jpg 3KB
animate_face_497_ema.jpg 8KB
neu160_1.jpg 2KB
neu160_4.jpg 3KB
Python.png 6KB
nrsd_217_ema.jpg 4KB
neu160_0.jpg 7KB
nrsd_194_ema.jpg 4KB
cifar_244_ema.jpg 11KB
nrsd_298_ema.jpg 4KB
animate_face_428_ema.jpg 9KB
nrsd_276_ema.jpg 4KB
LICENSE 11KB
webui
__init__.py 144B
web.py 26KB
utils
initializer.py 9KB
utils.py 8KB
__init__.py 144B
checkpoint.py 7KB
logger.py 2KB
lr_scheduler.py 1KB
datasets
dataset_demo
class_1
image_1.jpg 10KB
image_2.jpg 9KB
class_3
image_1.jpg 19KB
image_2.jpg 20KB
class_2
image_1.jpg 15KB
image_2.jpg 18KB
README.md 935B
model
__init__.py 144B
networks
__init__.py 144B
unet.py 5KB
base.py 2KB
sr
srv1.py 3KB
cspdarkunet.py 5KB
samples
__init__.py 144B
ddim.py 4KB
ddpm.py 4KB
base.py 4KB
modules
__init__.py 144B
ema.py 2KB
module.py 4KB
activation.py 966B
block.py 5KB
attention.py 2KB
conv.py 3KB
requirements.txt 146B
test
test_module.py 9KB
__init__.py 144B
noising_test
landscape
noising_test.jpg 41KB
noise
noise.jpg 1.63MB
.gitignore 3KB
results
README.md 29B
sr
dataset.py 3KB
interface.py 4KB
train.py 17KB
demo.py 3KB
共 84 条
- 1
资源评论
Java程序员-张凯
- 粉丝: 1w+
- 资源: 6723
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功