Detecting multiple objects using Tensorflow 2.0

I am following this tutorial trying to get object detection working. https://github.com/tensorflow/models/blob/master/research/object_detection/colab_tutorials/eager_few_shot_od_training_tf2_colab.ipynb It works great but I want to detect multiple classes (3). The tutorial states

for simplicity, we assume a single class in this colab; though it should be straightforward to extend this to handle multiple classes

but I can find a "straightforward" way to extend it for multiple classes. Loading the images and its bounding boxes all work fine.

Add Comment
0 Answer(s)

Your Answer

By posting your answer, you agree to the privacy policy and terms of service.