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

训练自己的数据集 #43

Closed
qhykwsw opened this issue Feb 12, 2019 · 5 comments
Closed

训练自己的数据集 #43

qhykwsw opened this issue Feb 12, 2019 · 5 comments

Comments

@qhykwsw
Copy link

qhykwsw commented Feb 12, 2019

谢谢大佬的贡献,我向请问下如果把自己的数据集制作成标准的VOC格式,然后要用YOLOV3进行训练。
目前我能想到的修改有:
(1)首先改动yolov3.yml中的“labels”和“data_root_dir”以及一些超参数,
(2)还有在_yolov3.py修改
num_classes=数据集对应的种类数
想问下还需要改动什么么?

另外,mobilenet, mobilenetv2, shufflenet(g2), shufflenetv2(1x), squeezenext(1.0-SqNxt-23v5), light xception, xception这些模型的权重文件可否提供一下呢?

@mileistone
Copy link
Contributor

mileistone commented Feb 13, 2019

I think you should modify the following:
1, labels, data_root_dir and other hyper parameters in yolov3.yml
2, if you need, you may modify anchors, thresh in loss

You don't need modify the num_classes which can be deduced from labels in yolov3.yml

I think you don't need weights of other models , you can just train from scratch, it's easy to converge.

@qhykwsw
Copy link
Author

qhykwsw commented Feb 13, 2019

Thank your for your reply. I have trained the model on my own dataset.

@hh7383
Copy link

hh7383 commented Mar 11, 2019

Thank your for your reply. I have trained the model on my own dataset.

我也在训练自己的数据集,有些问题想向你请教,请问你的数据集怎么制作放置的为什么我按照VOC2007的数据放置且按照readme的设置后,在执行python examples/train.py Yolov2时出现了FileNotFoundError: [Errno 2] No such file or directory: '/VOCdevkit/train/JPEGImages/0112.jpg'的问题,train文件目录是怎么来的?我并没有在VOCdevkit中创建过train文件夹。

@chakkritte
Copy link

@qhykwsw Could you tell me about your own datasets are VOC format (XML) or YOLO format (txt)?

Thank you.

@Aurorayy
Copy link

Thank your for your reply. I have trained the model on my own dataset.

我也在训练自己的数据集,有些问题想向你请教,请问你的数据集怎么制作放置的为什么我按照VOC2007的数据放置且按照readme的设置后,在执行python examples/train.py Yolov2时出现了FileNotFoundError: [Errno 2] No such file or directory: '/VOCdevkit/train/JPEGImages/0112.jpg'的问题,train文件目录是怎么来的?我并没有在VOCdevkit中创建过train文件夹。

请问您解决了吗

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

5 participants