8.16打卡 DAY43 复习日

作业 :kaggle找到一个图像数据集,用cnn网络进行训练并且用grad-cam做可视化

进阶:并拆分成多个文件

选择Intel Image Classification作为数据集:

  • 含有6类自然场景:建筑、森林、冰川、山、海洋、街道
  • 已经分好了train和test

几个注意点:

  1. 注意数据集下载时的结构,需要去掉一层文件
  2. 对于图片数量较少的类别,比如冰川,需要用数据增强平衡数据集

拆分文件目录如下:

python 复制代码
project/
├── data/
│   ├── train/   
│   └── val/   
├── src/
│   ├── model.py
│   ├── train.py
│   ├── dataset.py
│   └── gradcam_vis.py
├── outputs/
│   ├── best.pth
│   └── cam_examples/
├── requirements.txt
└── README.md

@浙大疏锦行