search

thtrieu / darkflow

Translate darknet to tensorflow. Load trained weights, retrain/fine-tune using tensorflow, export constant graph def to mobile devices
GNU General Public License v3.0
6.13k stars 2.09k forks source link

How to put a darkflow model into Android Studio #1197

Open M1nseoPark opened 3 years ago

M1nseoPark commented 3 years ago

I created an object detection model that I trained through darkflow. and I changed this to the tflite format to put into Android Studio

but I when I try to add metadata to this file, the following error occurs: ValueError: The number of output tensors (1) should match the number of output tensor metadata (4)

For reference, the code for adding metadata was based on the below site. (https://www.tensorflow.org/lite/convert/metadata)

I want to know if there is any other way to add metadata to the darkflow model or if my model or code is wrong :sob: