Closed JuanSeBestia closed 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.
Fixed in v0.2.2.
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