通过深度残差网络ResNet进行图像分类(pytorch网络多网络集成配置)
通过深度残差网络进行图像分类(pytorch网络多网络集成配置)简介本项目通过配置文件修改,实现pytorch的ResNet18, ResNet34, ResNet50, ResNet101, ResNet152网络更替,并通过代码实现自动生成识别所需的标签文件classes.txt(默认使用编码utf-8)。开发者只需要填写一些基本的元素如数据集地址,图像预处理大小,模型保存地址即可实现模型的训练和调用。(GitHub: https://github.com/CaoYuGang/ResNet_pyt
原创
2020-12-15 14:19:41 ·
9377 阅读 ·
6 评论