ctw-baseline icon indicating copy to clipboard operation
ctw-baseline copied to clipboard

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

Open xmq1221 opened this issue 6 years ago • 2 comments

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

xmq1221 avatar Mar 29 '18 07:03 xmq1221

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].

yuantailing avatar Mar 29 '18 07:03 yuantailing

Thanks, I will try it immediately

xmq1221 avatar Mar 29 '18 08:03 xmq1221