search

tryolabs / luminoth

Deep Learning toolkit for Computer Vision.
https://tryolabs.com
BSD 3-Clause "New" or "Revised" License
2.4k stars 400 forks source link

[train] WARNING:tensorflow:Function draw_object_prediction failed with KeyError. Key value: 'classification_prediction' #235

Closed JuanSeBestia closed 5 years ago

JuanSeBestia commented 5 years ago

The warning appears whenever the checkpoint is saved.

WARNING:tensorflow:Function draw_object_prediction failed with KeyError. Key value: 'classification_prediction'

I have not taken care because it does not affect the result or I think so

Mybe you need my config config_ssd_colab_201811 (copy).txt

nagitsu commented 5 years ago

It's a bug with the SSD Tensoroard image visualization, will look into it.

If you want to avoid the error for now, you can either turn debug mode on (which will log all the debugging images for SSD on Tensorboard) or disable image visualization by emptying the image_vis field in the config.

dekked commented 5 years ago

Fixed in v0.2.2.