thanks for your contribution, making the code compatible with TF2 is a good idea.
However, the script (python main.py ) crashes with your modifications because the method inferBatch must return a tuple of length 2 (second value is the probability).
Please correct this, then I will merge your changes.
Hi,
thanks for your contribution, making the code compatible with TF2 is a good idea. However, the script (python main.py ) crashes with your modifications because the method inferBatch must return a tuple of length 2 (second value is the probability).
Please correct this, then I will merge your changes.
Best Harald