search

yizt / Grad-CAM.pytorch

pytorch实现Grad-CAM和Grad-CAM++,可以可视化任意分类网络的Class Activation Map (CAM)图,包括自定义的网络;同时也实现了目标检测faster r-cnn和retinanet两个网络的CAM图;欢迎试用、关注并反馈问题...
Apache License 2.0
730 stars 167 forks source link

加载自己的模型参数 #13

Open Dtappledoghuati opened 4 years ago

Dtappledoghuati commented 4 years ago

请问如何在main.py文件中加载自己的模型及参数，我所使用的模型不是pytorch官方提供的，谢谢

yizt commented 4 years ago

@Dtappledoghuati 在get_net(net_name, weight_path=None)函数中，把自己的网络构建添加进去就好

Dtappledoghuati commented 4 years ago

谢谢，已成功！

haobabuhaoba commented 3 years ago

谢谢，已成功！

请问你是怎么添加的，能给份代码参考一下吗，我也想试试自己的

haobabuhaoba commented 3 years ago

谢谢，已成功！

993210357@qq.com 我的邮箱，谢谢你

nnliuheart commented 3 years ago

谢谢，已成功！

993210357@qq.com 我的邮箱，谢谢你

请问你你解决了吗，我也想知道该怎么做，1092931084@qq.com

nnliuheart commented 3 years ago

谢谢，已成功！ 你好，请问你是怎么添加的，我也需要，我也想试试自己的 1092931084@qq.com 我的邮箱，谢谢你