TFDetection icon indicating copy to clipboard operation
TFDetection copied to clipboard

convert to tflite

Open avber opened this issue 2 years ago • 3 comments

Hi,

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

Thanks

avber avatar Sep 22 '22 09:09 avber

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

Burf avatar Sep 28 '22 14:09 Burf

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

avber avatar Oct 26 '22 04:10 avber

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

Burf avatar Mar 09 '23 01:03 Burf