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

Why didn't I generate complete chinese.x.txt #9

Open
xmq1221 opened this issue Mar 29, 2018 · 2 comments
Open

Why didn't I generate complete chinese.x.txt #9

xmq1221 opened this issue Mar 29, 2018 · 2 comments

Comments

@xmq1221
Copy link

xmq1221 commented Mar 29, 2018

Hello, I used your trained yolo model and found a missing chinese.names file,then I download ctw-trainval-01-of-26.tar,used prepare_train_data.py to generate chinese.names.

I ran prepare_test_data.py with all provided test data,got these

root@0a0fe4b86df1:~/shared/ctw-baseline/detection/products# ls
backup           chinese.2.data  chinese.8.data         test         test.3.txt  test.9.txt
cates.json       chinese.3.data  chinese.9.data         test.0.txt   test.4.txt  test.txt
chinese.0.data   chinese.4.data  chinese.data           test.1.txt   test.5.txt  trainval
chinese.1.data   chinese.5.data  chinese.names          test.10.txt  test.6.txt  trainval.txt
chinese.10.data  chinese.6.data  darknet19_448.conv.23  test.11.txt  test.7.txt  yolo-chinese-test.cfg
chinese.11.data  chinese.7.data  results                test.2.txt   test.8.txt  yolo-chinese.cfg

Then about more than a dozen hours after I ran eval.py,I got these txt in /detection/products/result/

root@0a0fe4b86df1:~/shared/ctw-baseline/detection/products/results# ls
chinese.10.txt  chinese.11.txt  chinese.3.txt  chinese.4.txt  chinese.6.txt  chinese.7.txt  chinese.8.txt  chinese.9.txt

,it didn't generate complete chinese.x.txt,I wanna know why

@yuantailing
Copy link
Owner

yuantailing commented Mar 29, 2018

It seems like out of GPU memory, died threads output nothing.
per GPU memory ≈ 3.8 GB * num_thread (Line 74, eval.py) / TEST_NUM_GPU (line 30, settings.py)
Please modify these two variables.

You don't have to rerun all tests, just replace range(settings.TEST_SPLIT_NUM) (Line 74, eval.py) with you missing outputs [0, 1, 2, 5].

@xmq1221
Copy link
Author

xmq1221 commented Mar 29, 2018

Thanks, I will try it immediately

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

2 participants