Readme

项目结构

dogs-vs-cats/
│
├── data/                  # 保存解压后的图片数据
│   └── train/
│
├── dataset.py             # 自定义数据集
├── model.py               # 定义模型结构
├── train.py               # 训练脚本
└── infer.py               # 推理脚本
│
└── predict/               # 推理集