search

Thinklab-SJTU / R3Det_Tensorflow

Code for AAAI 2021 paper: R3Det: Refined Single-Stage Detector with Feature Refinement for Rotating Object
Apache License 2.0
540 stars 122 forks source link

训练次数问题 #87

Closed huyu6789 closed 3 years ago

huyu6789 commented 3 years ago

作者你好，请问使用你们的模型在训练自己的数据集的时候，至少需要训练多少次才能达到一个比较好的效果呢，因为某些原因时间有限，如果按照您repo中示例的那样训练100W次，时间来不及(2080TI训练200k次大概需要40个小时)。关于这一点，您有什么好的建议吗，现在最高训练到200k次，置信度阈值调到0.001也没有任何输出框和坐标。 自己的数据集介绍:2000张图片，分辨率都为1024x1024(使用data_crop.py进行裁剪和数据转换过)，图片格式.png，标注保持和DOTA数据集一致。 ps:附训练40k次的loss图和图片。 1 2

shizhuoyang commented 3 years ago

我也遇到了这个问题，请问你解决了吗？

yangxue0827 commented 3 years ago

已通过微信解决

AITTSMD commented 3 years ago

作者好，我也遇到类似的问题。我训练了17w次，80多轮，输出结果置信度很低，位置也不准确。