VisDrone icon indicating copy to clipboard operation
VisDrone copied to clipboard

Published 20 hours ago • verified publisher icon Mhaiyang

Training model

Open ZhihongChen123 opened this issue 6 years ago • 2 comments

I tried to train the model, but the rpn_bbox_loss remained at around 700 and the model could not converge.Please help me, thank you.

ZhihongChen123 avatar May 22 '18 06:05 ZhihongChen123

Hi, I had the same issue above. Could you please help us with it?

Haotian-Zhang avatar Oct 27 '18 08:10 Haotian-Zhang

@ZhihongChen123 @Haotian-Zhang did you succeeded in training the model? I run train_visdrone.py. After both 1, 5, and 20 epochs the model is no longer able to detect anything. I can only guess, what is the reason:

  1. There is a bug in the code.
  2. All weights of the model have to be fine-tuned (commented block of code)
  3. Due to the constant learning rate, the optimization doesn't coverage. I'll try something like ReduceLROnPlateau in my further attempts.

@Mhaiyang please, share some more details of the training process?

katyamineeva avatar May 05 '19 17:05 katyamineeva