Open SubratoChakravorty opened 4 years ago
The predict.py'' takes care of both Pascal and VOC evaluations. Please check if the FLAG
eval_coco_on_voc'' is correctly set. The ``dining
table'' appears only in the coco vocabulary
https://github.com/yekeren/Cap2Det/blob/cap2det/data/coco_label.txt but
not the VOC one
https://github.com/yekeren/Cap2Det/blob/cap2det/data/voc_label.txt.
On Fri, Mar 20, 2020 at 7:49 PM SubratoChakravorty notifications@github.com wrote:
Hi Keren,
I was able to train a cap2det model, but while running predict.py I get this error, Traceback (most recent call last): File "train_new/predict.py", line 634, in tf.app.run() File "/subrato-volume/tfoffenv/lib/python3.6/site-packages/tensorflow_core/python/platform/app.py", line 40, in run _run(main=main, argv=argv, flags_parser=_parse_flags_tolerate_undef) File "/subrato-volume/tfoffenv/lib/python3.6/site-packages/absl/app.py", line 300, in run _run_main(main, args) File "/subrato-volume/tfoffenv/lib/python3.6/site-packages/absl/app.py", line 251, in _run_main sys.exit(main(argv)) File "train_new/predict.py", line 597, in main categories) File "train_new/predict.py", line 386, in _run_evaluation for x in groundtruth_classes[:num_groundtruths] File "train_new/predict.py", line 386, in for x in groundtruth_classes[:num_groundtruths] KeyError: 'dining table'
— You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub https://github.com/yekeren/Cap2Det/issues/21, or unsubscribe https://github.com/notifications/unsubscribe-auth/AA6CPAKJIM2CSXJIJG5AWZTRIP6JLANCNFSM4LQWUMWQ .
-- Thanks, best regards.
Keren
Hi Keren,
I was able to train a cap2det model, but while running predict.py I get this error, Traceback (most recent call last): File "train_new/predict.py", line 634, in
tf.app.run()
File "/subrato-volume/tfoffenv/lib/python3.6/site-packages/tensorflow_core/python/platform/app.py", line 40, in run
_run(main=main, argv=argv, flags_parser=_parse_flags_tolerate_undef)
File "/subrato-volume/tfoffenv/lib/python3.6/site-packages/absl/app.py", line 300, in run
_run_main(main, args)
File "/subrato-volume/tfoffenv/lib/python3.6/site-packages/absl/app.py", line 251, in _run_main
sys.exit(main(argv))
File "train_new/predict.py", line 597, in main
categories)
File "train_new/predict.py", line 386, in _run_evaluation
for x in groundtruth_classes[:num_groundtruths]
File "train_new/predict.py", line 386, in
for x in groundtruth_classes[:num_groundtruths]
KeyError: 'dining table'
I am using the single trainer file train.sh to run training and evaluation.