-
-
Notifications
You must be signed in to change notification settings - Fork 17.2k
Description
Search before asking
- I have searched the YOLOv5 issues and discussions and found no similar questions.
Question
I'm training a model and I keep getting WARNING that weren't generated before. WARNING persists even after epoch goes over 400.
Can I just keep to learn?
Changes
Does not occur before changes are applied. ( WARNING: NMS time limit 10.0s exceeded )
-
Changed object detection from two(pics > A:1800, B:1000) to three(pics > A:1800, B1000, C:500 ).
1-1. The newly created class C has a WARNING( WARNIGN: corrupt JPEG restored and saved ) image at 25 train val each. -
fl_gamma: 0.0 --> flgamma: 1.5
-
degree: 0.0 --> 20
-
mixup:0.0 --> 0.3
error
Epoch gpu_mem box obj cls labels img_size
61/449 29.5G 0.03907 0.003123 0.002231 664 640: 100%|██████████| 11/11 [00:48<00:00, 4.44s/it]
Class Images Labels P R [email protected] [email protected]:.95: 0%| | 0/2 [00:00<?, ?it/s]
WARNING: NMS time limit 10.0s exceeded
Class Images Labels P R [email protected] [email protected]:.95: 50%|▌| 1/2 [00:13<00:13, 13.23
WARNING: NMS time limit 10.0s exceeded
Class Images Labels P R [email protected] [email protected]:.95: 100%|█| 2/2 [00:25<00:00, 12.78
all 687 706 0.687 0.128 0.121 0.0646
Epoch gpu_mem box obj cls labels img_size
62/449 29.5G 0.03979 0.003183 0.002103 687 640: 100%|██████████| 11/11 [00:17<00:00, 1.62s/it]
Class Images Labels P R [email protected] [email protected]:.95: 0%| | 0/2 [00:00<?, ?it/s]
WARNING: NMS time limit 10.0s exceeded
Class Images Labels P R [email protected] [email protected]:.95: 50%|▌| 1/2 [00:11<00:11, 11.91
WARNING: NMS time limit 10.0s exceeded
Class Images Labels P R [email protected] [email protected]:.95: 100%|█| 2/2 [00:24<00:00, 12.36
all 687 706 0.487 0.216 0.205 0.122
Epoch gpu_mem box obj cls labels img_size
63/449 29.5G 0.03688 0.003107 0.002027 626 640: 100%|██████████| 11/11 [00:42<00:00, 3.90s/it]
Class Images Labels P R [email protected] [email protected]:.95: 0%| | 0/2 [00:00<?, ?it/s]
WARNING: NMS time limit 10.0s exceeded
Class Images Labels P R [email protected] [email protected]:.95: 50%|▌| 1/2 [00:12<00:12, 12.34
WARNING: NMS time limit 10.0s exceeded
Additional
train code
python train.py --img 640 --batch 256 --epochs 450 --data data/custom2.yaml --cfg models/yolov5n.yaml --weights yolov5n.pt --workers 16 --name goral_color_nano_mixup0.3 --patience 300
NOW hyp.yaml
lr0: 0.01
lrf: 0.1
momentum: 0.937
weight_decay: 0.0005
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: 1.5
hsv_h: 0.015
hsv_s: 0.7
hsv_v: 0.4
degrees: 20.0
translate: 0.1
scale: 0.5
shear: 0.0
perspective: 0.0
flipud: 0.0
fliplr: 1.0
mosaic: 1.0
mixup: 0.3
copy_paste: 0.0