search

yangxue0827 / RotationDetection

This is a tensorflow-based rotation detection benchmark, also called AlphaRotate.
https://rotationdetection.readthedocs.io/
Apache License 2.0
1.08k stars 182 forks source link

您好，请问如果我需要更换loss的话比如Modulated Loss；检测器是retinanet。是否是更改sfgs.py里面的REG_LOSS_MODE = None #46

Closed homxxx closed 3 years ago

yangxue0827 commented 3 years ago

你直接用rsdet不就是retinanet+ml吗

homxxx commented 3 years ago

好的 请问如果想在ucas-aod数据集上面跑这两个方法，应该如何配置？我看ucas-aod没有rsdet的cfg文件。

homxxx commented 3 years ago

这个项目支持batch size >1嘛？

yangxue0827 commented 3 years ago

基于retinanet的单阶段方法是支持的，参考tools/retinanet下的train_batch.py

yangxue0827 commented 3 years ago

ucas-aod配置文件参考dota的改一下就行了

homxxx commented 3 years ago

好的

homxxx commented 3 years ago

请问应该如何构建自己的检测器，比如darknet+fpn/bifpn ...

yangxue0827 commented 3 years ago

这肯定需要你看明白代码的结构，然后拼装了

homxxx commented 3 years ago

好的 谢谢。请问repo里面有基于darknet的现成检测器嘛?

yangxue0827 commented 3 years ago

如果你只是换一下backbne的话，只要吧cfgs.py里的NET_NAME设置成'darknet'就行了，记得提前下载darknet的预训练权重

homxxx commented 3 years ago

嗯嗯 好的