Hiyuur / SR-TOD

This is the official code for the paper Visible and Clear: Finding Tiny Objects in Difference Map.
Apache License 2.0
8 stars 1 forks source link

文章中核心代码 #1

Open SmoothJing opened 1 month ago

SmoothJing commented 1 month ago

作者,您好 您的工作非常好。 但是我看到您公开整理的代码真的很多, 想请教下关于文章中模型的核心代码在哪个脚本下呢 特别期待得到您的回复 谢谢!

Hiyuur commented 1 month ago

感谢你的认可。项目在srtod_project中,以srtod_cascade_rcnn为例,核心代码在srtod_cascadercnn.py中。其他文件的作用主要是重写一下数据管道,增加原始图像输入。实际上也可以直接在srtod_cascade_rcnn.py中增加denorm的操作还原出原始输入图像,这样虽然不需要重写数据管道,但是会增加重复的不必要计算并且违反mmdetection框架将数据处理和后续模块解耦的原则,影响复用。

SmoothJing commented 1 month ago

感谢你的认可。项目在srtod_project中,以srtod_cascade_rcnn为例,核心代码在srtod_cascadercnn.py中。其他文件的作用主要是重写一下数据管道,增加原始图像输入。实际上也可以直接在srtod_cascade_rcnn.py中增加denorm的操作还原出原始输入图像,这样虽然不需要重写数据管道,但是会增加重复的不必要计算并且违反mmdetection框架将数据处理和后续模块解耦的原则,影响复用。

首先,感谢您的热心回复! 这让我对代码结构清晰很多。 但是我还有一个小问题,我想要在srtod_cascadercnn.py这个脚本中实例化, 以便于我能够debug进行跟踪代码,但是方向差异的原因,我不太清楚咱们的 if name == 'main':需要送写怎么样的参数,请问您方便发下实例化使得 我可以更好的调试代码嘛? 再次感谢! 期待您的回复!