yijingru / BBAVectors-Oriented-Object-Detection

[WACV2021] Oriented Object Detection in Aerial Images with Box Boundary-Aware Vectors
MIT License
462 stars 87 forks source link

loss下降不下去 #56

Open AIFANLIN opened 3 years ago

AIFANLIN commented 3 years ago

你好,我用您的模型对身份证进行定位,其中正面身份证大约1千张,背面身份证大约200张,就只有这两类。模型迭代80个epoch左右,loss就在3-4之间震荡,降不下去,请问可能的原因是什么?

jaychen1996-byte commented 3 years ago

我遇到类似的问题,我也是训练两类,但是我的图片只有5张,大小为(640,480),训练速度很慢,loss不收敛,80个epoch在loss7-8间震荡,尽管我调整了lr_scheduler仍然没有见效。