注意,本文只提供学习的思路,严禁违反法律以及破坏信息系统等行为,本文只提供思路
如有侵犯,请联系作者下架
九宫格分类如下
这种就是最简单的分类识别了,用迁移学习resnet训练即可,下面来看成品
训练代码查看往期文章中就有,部分代码如下:
DEVICE = torch.device('cuda:0') # 使用显卡
model = models.resnet50
注意,本文只提供学习的思路,严禁违反法律以及破坏信息系统等行为,本文只提供思路
如有侵犯,请联系作者下架
九宫格分类如下
这种就是最简单的分类识别了,用迁移学习resnet训练即可,下面来看成品
训练代码查看往期文章中就有,部分代码如下:
DEVICE = torch.device('cuda:0') # 使用显卡
model = models.resnet50