pjreddie / darknet

Convolutional Neural Networks
http://pjreddie.com/darknet/
Other
25.81k stars 21.33k forks source link

I try to training custom dataset by using colab.I found some problems with output after training #1600

Open NoraJiang opened 5 years ago

NoraJiang commented 5 years ago

Here is the article i followed:

http://bennycheung.github.io/yolo-for-real-time-food-detection Once I training the customized Darknet’s YOLO by "!./darknet detector train data/food100/food100.data data/food100/yolov2-food100.cfg data/darknet19_448.conv.23> data/food100/train.log"

The output :

..... 15 conv 256 1 x 1 / 1 26 x 26 x 512 -> 26 x 26 x 256 0.177 BF 16 conv 512 3 x 3 / 1 26 x 26 x 256 -> 26 x 26 x 512 1.595 BF 17 max 2 x 2 / 2 26 x 26 x 512 -> 13 x 13 x 512 0.000 BF 18 conv 1024 3 x 3 / 1 13 x 13 x 512 -> 13 x 13 x1024 1.595 BF 19 conv 512 1 x 1 / 1 13 x 13 x1024 -> 13 x 13 x 512 0.177 BF 20 conv 1024 3 x 3 / 1 13 x 13 x 512 -> 13 x 13 x1024 1.595 BF 21 conv 512 1 x 1 / 1 13 x 13 x1024 -> 13 x 13 x 512 0.177 BF 22 conv 1024 3 x 3 / 1 13 x 13 x 512 -> 13 x 13 x1024 1.595 BF 23 conv 1024 3 x 3 / 1 13 x 13 x1024 -> 13 x 13 x1024 3.190 BF 24 conv 1024 3 x 3 / 1 13 x 13 x1024 -> 13 x 13 x1024 3.190 BF 25 route 16 26 conv 64 1 x 1 / 1 26 x 26 x 512 -> 26 x 26 x 64 0.044 BF 27 reorg / 2 26 x 26 x 64 -> 13 x 13 x 256 28 route 27 24 29 conv 1024 3 x 3 / 1 13 x 13 x1280 -> 13 x 13 x1024 3.987 BF 30 conv 525 1 x 1 / 1 13 x 13 x1024 -> 13 x 13 x 525 0.182 BF 31 detection mask_scale: Using default '1.000000' Total BFLOPS 29.510 Allocate additional workspace_size = 49.84 MB Loading weights from data/darknet19_448.conv.23...Done! Unable to init server: Could not connect: Connection refused

(average loss:13433): Gtk-WARNING **: 06:21:22.690: cannot open display:

And the train.log:

yolov2-food100

seen 32 Learning Rate: 0.001, Momentum: 0.9, Decay: 0.0005 If error occurs - run training with flag: -dont_show

I can't get any logs looks like : Region Avg IOU: 0.727497, Class: 0.273119, Obj: 0.348449, No Obj: 0.004568, Avg Recall: 0.875000, count: 8 Region Avg IOU: 0.767684, Class: 0.285230, Obj: 0.376508, No Obj: 0.003249, Avg Recall: 0.875000, count: 8 9998: 5.413931, 5.206974 avg, 0.000010 rate, 2.222207 seconds, 639872 images

Any help would be very appreciated!!

NoraJiang commented 5 years ago

I add '-dont_show', now it worked....

tanluren commented 5 years ago

colab has deactivated GUI ,you can deactivate the relative function of OPENCV or just simply set OPENCV 0 when you make.

khushisinhax commented 2 years ago

I add '-dont_show', now it worked....

Where did you add it? Im facing the same issue

Lucas-R-Garcia commented 2 years ago

Eu adiciono '-dont_show', agora funcionou ....

Onde você adicionou? estou enfrentando o mesmo problema

Se você descobrir me avise, estou com o mesmo problema