单张GPU能跑出效果来吗。

yangxue0827 / RotationDetection

This is a tensorflow-based rotation detection benchmark, also called AlphaRotate.

https://rotationdetection.readthedocs.io/

Apache License 2.0

1.08k stars 181 forks source link

Closed jiangfeizi closed 2 years ago

jiangfeizi commented 2 years ago

我单张gpu在R3Det，DOTA数据集上跑了20w步基本没效果。

yangxue0827 commented 2 years ago

r3det单张gpu好像不行

yangxue0827 commented 2 years ago

r3det单张gpu好像不行

jiangfeizi commented 2 years ago

谢谢回复，这个benchmark中有哪个网络适合单张GPU训练的呢

yangxue0827 commented 2 years ago

retinanet是可以的，我们也提供了pytorch版本的r3det，可以用单张卡跑，https://github.com/SJTU-Thinklab-Det/r3det-pytorch

jiangfeizi commented 2 years ago

好的，我去试试，谢谢回复