search

kinhong / OpenLabeler

OpenLabeler is an open source desktop application for annotating objects for AI appplications
Apache License 2.0
114 stars 11 forks source link

Any plans to support Tensorflow 2? #17

Closed veonua closed 3 years ago

veonua commented 4 years ago

Object detection API has been updated to run on the latest TF version, https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/tf2_detection_zoo.md

it seems like output vector has been replaced with a dictionary, at least in my code I had to replace 

num_detections = result[0].numpy()
classes = result[1].numpy()

to 

num_detections = result["num_detections"].numpy()
classes = result['detection_classes'].numpy()
kinhong commented 4 years ago

@veonua I am still waiting for a more official release of TensorFlow 2.0 Java API (I have been following this issue)

veonua2 commented 4 years ago

References: https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-2.3.0.jar https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-cpu-linux-x86_64-2.3.0.tar.gz https://www.tensorflow.org/install/lang_java

looks promising, but I am not fully aware of set of jars required to run it on Java, I assume update of protobuf would be good enough to run inference.

kinhong commented 3 years ago

@veonua2 In release v1.3.0, OpenLabeler now supports TensorFlow 2.3.1.

veonua commented 3 years ago

thank you very much