mmdetection-annotated icon indicating copy to clipboard operation
mmdetection-annotated copied to clipboard

模型可视化

Open shuxp opened this issue 5 years ago • 9 comments

你好,你的模型可视化用的什么工具?能否分享下?谢谢~

shuxp avatar Jun 17 '19 15:06 shuxp

Netron

ming71 avatar Jun 20 '19 02:06 ming71

Netron

我试了,但是对于检测任务都只会出现下面独立模块,不知道你那边存储模型的时候是否做了特殊处理呢?

image

shuxp avatar Jun 20 '19 06:06 shuxp

Netron只能识别继承自nn.Module的特定类。你这个结果已经比较详细了,还有什么问题吗?或者说你想要的怎样的模型?

ming71 avatar Jun 20 '19 07:06 ming71

我的模型是用https://github.com/jwyang/faster-rcnn.pytorch 代码训练出来的,自己改了些东西,用Netron打开就是上面那样,我看了下Netron的源码,可惜electron不熟。。

你能帮忙提点意见嘛

shuxp avatar Jun 20 '19 10:06 shuxp

你想要什么样的结构啊...我看你可视化的已经很详细了啊

ming71 avatar Jun 23 '19 04:06 ming71

看不到箭头,不知道数据流流向以及各个地方的输出维度。。

shuxp avatar Jun 25 '19 11:06 shuxp

作者你好,关注你这个github很久了,能不能加你一个联系方式

nuist-xinyu avatar Jul 02 '19 06:07 nuist-xinyu

作者你好,关注你这个github很久了,能不能加你一个联系方式

邮箱[email protected],可以邮件联系

ming71 avatar Jul 10 '19 01:07 ming71

看不到箭头,不知道数据流流向以及各个地方的输出维度。。

额,如果netron不行的话,还有一个更好用的工具keras的自带可视化模块。找一个相同网络的keras实现版本,用自带可视化模块输出模型结构就行。希望能帮到你==

ming71 avatar Aug 27 '19 01:08 ming71