本教程跟随B站UP:同济子豪兄 ,配套的Github代码:https://github.com/TommyZihao/Train_Custom_Dataset
本repo内容源自子豪兄,且加上自己的注释和实践进行了重构,本repo支持子豪兄一切知识权益,侵权必删!
本repo内容属于人工智能实践内容,学习理论是一方面,实践也要跟上哦!目前本repo包含的内容有:
- 图像分类 ;
- ...
- 1、图像分类
- 1.1、构建自己的图像分类数据集
- 1.2、预训练图像分类模型的预测
- 1.3、迁移学习训练自己的图像分类模型
- 1.4、训练得到的模型预测图像、视频、摄像头实时画面
- 1.5、在测试集上评估图像分类算法精度
- 1.6、可解释性分析-CAM类激活热力图可视化
- 1.7、可解释性分析-Captum可视化
- ...