hizhangp / yolo_tensorflow

Tensorflow implementation of YOLO, including training and test phase.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

I detect nothing when using my own dataset to train.

ZoeSnow opened this issue · comments

I trained on my own dataset and replaced the yolo_small.ckpt with data,meta and index files, but no objects could be detected.(好桑心) Does anyone know what I should do?Thank you!!!

I trained on my own dataset and replaced the yolo_small.ckpt with data,meta and index files, but no objects could be detected.(好桑心) Does anyone know what I should do?Thank you!!!

I have the same problem, too. Did you solve it?
I trained it without yolo_small.ckpt, and the code run normally.

I haven't solved the error yet. My training process is also normal but I can't detect anything.

the loss layer in the training file seems to be wrong,so training on your your dataset can't work

@zuowk

Thank you for your reply. Did you train on your own dataset successfully? Could you tell me how to rectify the loss function?

the loss layer in the training file seems to be wrong,so training on your your dataset can't work

I trained it on the pascal_voc dataset without yolo_small.ckpt, and still can't detect anything, too.

commented

I have the same problem.Did anybody solve it?Or it means that i must use yolo_small.ckpt?

I found the reason!when i review the origin paper of yolo,i found that the auther pretrain the first 20 convonlutional layers on the ImageNet 1000-class dataset. So i pretrain the first conv-layers on another classification dataset(because i have no imagenet dateset),then train the whole network. Finally, it can detect things correctly! In short, if you want to train yolo from scratch, you should pretrain the first 20 conv-layers with a classification task