WongKinYiu / yolor

implementation of paper - You Only Learn One Representation: Unified Network for Multiple Tasks (https://arxiv.org/abs/2105.04206)
GNU General Public License v3.0
1.98k stars 524 forks source link

strange custom train batch image #252

Open jungyuko opened 2 years ago

jungyuko commented 2 years ago

Hello. I use yolor model for my object detection task. I trained my custom data with no argument anything. (no moasic, no flip, no translation, no rotation, ...)

I think train batch images must be clear images like this. train_batch4

But sometimes I got a strange batch images like this(about 10%). train_batch2

Can I help my problem?

I share my hyperparameter settings. lr0: 0.01 lrf: 0.2 momentum: 0.937 weight_decay: 0.001 warmup_epochs: 3.0 warmup_momentum: 0.8 warmup_bias_lr: 0.1 box: 0.05 cls: 0.5 cls_pw: 1.0 obj: 1.0 obj_pw: 1.0 iou_t: 0.2 anchor_t: 4.0 fl_gamma: 0.0 hsv_h: 0.0 hsv_s: 0.0 hsv_v: 0.0 degrees: 0.0 translate: 0.0 scale: 0.0 shear: 0.0 perspective: 0.0 flipud: 0.0 fliplr: 0.0 mosaic: 0.0 mixup: 0.0