search

AlexeyAB / darknet

YOLOv4 / Scaled-YOLOv4 / YOLO - Neural Networks for Object Detection (Windows and Linux version of Darknet )
http://pjreddie.com/darknet/
Other
21.57k stars 7.95k forks source link

使用cutmix=1增强后程序会挂掉 #5553

Open zys1994 opened 4 years ago

zys1994 commented 4 years ago

使用cutmix=1增强后训练一开始会挂掉，未提示任何错。请问作者知道这是什么问题吗？还是对我的数据库有什么要求，我用mosaic=1是可以完全训练完的。

zys1994 commented 4 years ago 
if (mosaic && cutmix) net->mixup = 4;
    else if (cutmix) net->mixup = 2;
    else if (mosaic) net->mixup = 3;

跟这一段代码有关吗？

tomingliu commented 4 years ago

5534

Tianyi-THU commented 4 years ago

使用cutmix=1增强后训练一开始会挂掉，未提示任何错。请问作者知道这是什么问题吗？还是对我的数据库有什么要求，我用mosaic=1是可以完全训练完的。

Hi, I am wondering you are using mosaic=1 with Classifier or Detector?