作业 :kaggle找到一个图像数据集,用cnn网络进行训练并且用grad-cam做可视化
进阶:并拆分成多个文件
选择Intel Image Classification作为数据集:
- 含有6类自然场景:建筑、森林、冰川、山、海洋、街道
- 已经分好了train和test
几个注意点:
- 注意数据集下载时的结构,需要去掉一层文件
- 对于图片数量较少的类别,比如冰川,需要用数据增强平衡数据集
拆分文件目录如下:
python
project/
├── data/
│ ├── train/
│ └── val/
├── src/
│ ├── model.py
│ ├── train.py
│ ├── dataset.py
│ └── gradcam_vis.py
├── outputs/
│ ├── best.pth
│ └── cam_examples/
├── requirements.txt
└── README.md