Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Question on training Yolo v1 #2

Open
leonardoaraujosantos opened this issue May 19, 2017 · 8 comments
Open

Question on training Yolo v1 #2

leonardoaraujosantos opened this issue May 19, 2017 · 8 comments

Comments

@leonardoaraujosantos
Copy link

Hi @ruiminshen, thanks for the project.

Could you show me how should I train for the yolo v1 (Tiny)?
Do I need to change the config.ini file? (I already downloaded the datasets VOC)

Thanks

@leonardoaraujosantos
Copy link
Author

What I'm trying to do....

  1. Download VOC 2007/2012, and extract at: Documents/Database/VOCdevki
  2. python3 cache.py -c config.ini config/yolo/tiny-20.ini -v
  3. python train.py -c config.ini config/yolo/tiny-20.ini

@leonardoaraujosantos
Copy link
Author

leonardoaraujosantos commented May 19, 2017

Hi @ruiminshen I think is working
train

Btw I'm documenting my findings about Yolo here:
https://leonardoaraujosantos.gitbooks.io/artificial-inteligence/content/single-shot-detectors/yolo.html

Do you accept a pull request for updating the Readme file?

@ruiminshen
Copy link
Owner

@leonardoaraujosantos Sorry, I'm a little busy these days and I didn't noticed your message.

Your operation is correct. To switch between models (YOLOv1 and YOLOv2), you can change the option "model" in "config" section in the default configuration file "config.ini", and the -c argument is used to override the options by another configuration file.

@xhzcyc
Copy link

xhzcyc commented Jun 6, 2017

When I trained it by following leonardoaraujosantos, I got this problem.
WARNING:tensorflow:Error encountered when serializing LAYER_NAME_UIDS.
Type is unsupported, or the types of the items don't match field type in CollectionDef.
'dict' object has no attribute 'name'

@TaihuLight
Copy link

TaihuLight commented Jun 25, 2017

@leonardoaraujosantos @xhzcyc @ruiminshen
Do you successfully train for YOLOv2 with VOC? And could you share your trained model? I try to train yolo2 using yolo-tf sevral times, but I always failed. @leonardoaraujosantos @xhzcyc
selection_006

selection_005

@zhyj3038
Copy link

@TaihuLight hi? can you get a accurate resuklt with yolo2?

@TaihuLight
Copy link

TaihuLight commented Aug 4, 2017

@zhyj3038 I can get the mAP with YOLO2, but multi-scale trianing and training with own dataset are need to be finished!

@ss199302
Copy link

@TaihuLight Hello, have you successfully implemented the test? Is the effect good?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

6 participants