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

Train的时候从epoch260开始,训练出来的class也不对 #2

Open
yueli822ms opened this issue Dec 8, 2020 · 0 comments
Open

Comments

@yueli822ms
Copy link

@DataXujing 您好,我下载了您给的weights进行训练,发现训练是从epoch 260开始的,而且即使我在data里面重新定义了class,训练出来的模型对应的还是coco class。能请您帮忙看看吗?谢谢
image

go_train_yolov4_sei.sh

python train.py
--device 0
--data ./data/sei_data.yaml
--cfg cfg/yolov4-pacsp-sei.cfg
--weights weights/yolov4-pacsp.pt
--img-size 1920 1920
--epochs 500
--batch-size 6
--resume False

sei_data.yaml

train and val datasets (image directory or *.txt file with image paths)

train: ./data/sei1202/images/train #
val: ./data/sei1202/images/val #
test: ./data/sei1202/images/test #

number of classes

nc: 2

class names

names: ['bottle', 'can']

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

1 participant