论文地址

https://arxiv.org/abs/1704.04861

运行环境

pytorch:1.10.0

数据集准备

下载猫狗分类数据集，放在dogvscat文件夹下。

下载链接：https://www.kaggle.com/competitions/dogs-vs-cats-redux-kernels-edition/data?select=sample_submission.csv

划分训练集验证集

python split_train_val.py

训练

python train.py

训练过程可视化

tensorboard --logdir='work_dir/runs'

产生提交结果

python generate_test_output.py

此模型产生的结果提交比赛端所得成绩十分奇怪，我也不知道为什么

训练结果可视化

python visual_result.py

训练过程