yekeren / Cap2Det

Implementation of our ICCV 2019 paper "Cap2Det: Learning to AmplifyWeak Caption Supervision for Object Detection"
Apache License 2.0
29 stars 9 forks source link

Error in Evaluation of Cap2Det model #21

Open SubratoChakravorty opened 4 years ago

SubratoChakravorty commented 4 years ago

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.

yekeren commented 4 years ago

The predict.py'' takes care of both Pascal and VOC evaluations. Please check if the FLAGeval_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