先参考一下这篇博客:https://www.jianshu.com/p/bf4a2082014f
我觉得可行!就应该使用softmax啊。
可行,但是目前没法输出每一类对应的概率,还是有点菜,后面有时间再补上吧。
pytorch---利用训练好的模型进行图片分类测试
最新推荐文章于 2024-03-26 00:01:41 发布
先参考一下这篇博客:https://www.jianshu.com/p/bf4a2082014f
我觉得可行!就应该使用softmax啊。
可行,但是目前没法输出每一类对应的概率,还是有点菜,后面有时间再补上吧。