ctw-baseline
ctw-baseline copied to clipboard
Why didn't I generate complete chinese.x.txt
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
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]
.
Thanks, I will try it immediately