Guillem96 / efficientdet-tf

Custom implementation of EfficientDet https://arxiv.org/abs/1911.09070
GNU General Public License v3.0
96 stars 23 forks source link

Bad prediction labels in the example notebook #12

Closed bessszilard closed 4 years ago

bessszilard commented 4 years ago

First of all, thanks for this library!

I ran the notebook which you provided, and on the algorithm finds "boat" and not "cat". My notebook is here: https://colab.research.google.com/drive/1-tmY7YxhCkvEvPXVExiMCmOyL4JZPTRf

I tried on another image, and instead of writing cars, it writes image

What causes the problem?

Guillem96 commented 4 years ago

Try it now, there was a commit that shouldn't be there. Be sure that you factory reset the colab runtime :)

bessszilard commented 4 years ago

Yes, fixed. Thanks.