day42_2025-08-16

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

进阶:并拆分成多个文件

选择Intel Image Classification作为数据集:

含有6类自然场景:建筑、森林、冰川、山、海洋、街道

已经分好了train和test

几个注意点:

注意数据集下载时的结构,需要去掉一层文件

对于图片数量较少的类别,比如冰川,需要用数据增强平衡数据集

拆分文件目录如下:

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

@浙大疏锦行