pytorch吧 关注:1,456贴子:2,583
  • 0回复贴,共1

萌新请教下,调用预训练网络返回的是数值,怎么对应上分类呢

只看楼主收藏回复


直接使用了官方的ResNet50预训练模型结构和参数,程序也能跑通,但是最后返回的分类是一个数值。个人感觉应该是分类对应的编号,想问下怎么能找到对应的是什么分类呢?


IP属地:北京1楼2020-06-05 17:47回复