哼º 2022-07-21 14:33 采纳率: 78.6%
浏览 65
已结题

深度学习预训练模型无法加载

在github上下载了用于人体姿态估计的堆叠沙漏网络,并且下载了作者提供的2HG的预训练模型
源码地址如下https://github.com/princeton-vl/pytorch_stacked_hourglass
但是在运行test.py时得到的结果如下

img

img

img

不知道是不是因为没有成功加载到预训练模型,如果想在test.py运行时载入预训练模型,应该怎么修改代码呢
现在我已经用作者的checkpoint.py替换了我自己训练的
(因为电脑显存问题自己训练的epoch为30,太小了,但自己训练完test时数值和这个也差不太多)

img

  • 写回答

1条回答 默认 最新

  • 乄洛尘 人工智能领域优质创作者 2022-07-21 18:05
    关注

    model = your_network
    model.load_state_dict(torch.load(model_weight_path, map_location=device))
    这只是个加载预训练权重的方法,具体的程序结合你的代码进行修改,CSDN你已开启陌生人防打扰,你虽然发消息给我,但无法私信联系。
    如果只是为了跑通代码,不如结合已跑通的代码进行一行一行调试,不积跬步无以至千里!

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论 编辑记录

报告相同问题?

问题事件

  • 系统已结题 7月31日
  • 已采纳回答 7月23日
  • 创建了问题 7月21日

悬赏问题

  • ¥20 搭建pt1000三线制高精度测温电路
  • ¥15 使用Jdk8自带的算法,和Jdk11自带的加密结果会一样吗,不一样的话有什么解决方案,Jdk不能升级的情况
  • ¥15 画两个图 python或R
  • ¥15 在线请求openmv与pixhawk 实现实时目标跟踪的具体通讯方法
  • ¥15 八路抢答器设计出现故障
  • ¥15 opencv 无法读取视频
  • ¥15 按键修改电子时钟,C51单片机
  • ¥60 Java中实现如何实现张量类,并用于图像处理(不运用其他科学计算库和图像处理库))
  • ¥20 5037端口被adb自己占了
  • ¥15 python:excel数据写入多个对应word文档