search

zhang-tao-whu / e2ec

E2EC: An End-to-End Contour-based Method for High-Quality High-Speed Instance Segmentation
Other
214 stars 45 forks source link

直接推理可视化混乱的情况 #38

Open yazheng0307 opened 9 months ago

yazheng0307 commented 9 months ago

使用Readme里面的visualize.py推理代码，预训练权重是作者提供的COCO权重，可视化结果为什么是这样的？换了其他图片和权重还是乱七八糟的线条 image image

zhang-tao-whu commented 9 months ago

抱歉，我测试了一下，可视化结果很正常。您是否移除了DCN或者使用了mmcv的DCN来替换了我代码中所采用的DCN，但是仍然加载的我所提供的权重？

yazheng0307 commented 9 months ago

抱歉，我测试了一下，可视化结果很正常。您是否移除了DCN或者使用了mmcv的DCN来替换了我代码中所采用的DCN，但是仍然加载的我所提供的权重？

是的，因为我这边DCN编译不成功，所以按照INSTALL.md的说明将cfg.model.use_dcn设置为False，加载作者提供权重。请问官方有没有非DCN的训练权重呢，我貌似没找到，是不是要自己训练一遍，感谢！

zhang-tao-whu commented 9 months ago

很抱歉没有提供去除DCN的权重，您可以试试用openmim安装mmcv-full，并使用mmcv提供的DCN，目前权重可以直接加载进去进行测试和可视化，结果显示没问题。如果去除DCN，需要您重新训练。

yazheng0307 commented 9 months ago

好的，谢谢大佬，我试试！

yazheng0307 commented 9 months ago

后续我按照你的方法，安装mmcv-full： pip install -U openmim mim install mmcv-full

尝试了几个mmcv-full的版本，并将dla.py的352行~355行代码改为： image

运行都报出以下键值对错误： image

请问你们用的mmcv-full是哪个版本呢？

zhang-tao-whu commented 9 months ago

清重新clone或者pull最新的commits，我修改了utils.py的62-66行。

zhang-tao-whu commented 9 months ago

我使用的mmcv-full版本为1.7.1.

yazheng0307 commented 9 months ago

好的，谢谢！