开发者社区

文档建议反馈控制台

文章/答案/技术大牛

发布

社区首页 >问答首页 >PyTorch针对哪些ImageNet类进行培训？

问PyTorch针对哪些ImageNet类进行培训？
EN

Stack Overflow用户

提问于 2020-02-27 03:41:02

回答 1查看 411关注 0票数 0

PyTorch允许您为分类here下载预先训练的模型，但不清楚如何从概率的输出张量返回到实际的类标签。有什么想法吗？

EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2020-02-27 03:52:27

请参阅此处的示例：https://discuss.pytorch.org/t/imagenet-classes/4923 https://www.learnopencv.com/pytorch-for-beginners-image-classification-using-pre-trained-models/

你需要下载imagenet_classes.txt

票数 0

EN

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/60421367

复制

相似问题