cifar100-with-data-augment

Using data augment in image classification

环境配置

python3.9

pytorch1.12.1

RTX3050Ti

具体参数见config.py

数据集会自动下载到文件夹cifar100下

python train.py --method {baseline/cutout/cutmix/mixup} --epoch300

python show_picture.py

结果保存在logs文件夹下

tensorboard --logdir=logs/{filename}

或者查看对应的txt文件

Name		Name	Last commit message	Last commit date
Latest commit History 10 Commits
logs		logs
README.md		README.md
config.py		config.py
cutout.py		cutout.py
resnet.py		resnet.py
show_picture.py		show_picture.py
train.py		train.py
utils.py		utils.py