search

Burf / TFDetection

A Detection Toolbox for Tensorflow2
Apache License 2.0
56 stars 18 forks source link

convert to tflite #3

Open avber opened 2 years ago

avber commented 2 years ago

Hi,

Is it possible to convert effdet_lite to the tflite format? If yes, could you add a code sample?

Thanks

Burf commented 2 years ago

First, please refer to function of tfdet.export.tflite. Usage will be updated soon.

avber commented 1 year ago

Hi @Burf ,

I have EffdetLite4 and MobileVit models trained on similar but not identical datasets. MobileVit model achieved more than 94% accuracy.

Could you show an example of how to replace classification head with the trained MobileVit model and export a combined model to tflite ?

The MobileVit model will need the original input image, not the preprocessed one. max_output_size, iou__threshold and score_threshold for bbox predictions can be hardcoded.

The intention is to use the combined model in tfjs.

Thanks

Burf commented 1 year ago

I don't know how the MobileVit structure or code is structured. Please refer to "usage/export/object_detection".