TFDetection
TFDetection copied to clipboard
convert to tflite
Hi,
Is it possible to convert effdet_lite to the tflite format? If yes, could you add a code sample?
Thanks
First, please refer to function of tfdet.export.tflite. Usage will be updated soon.
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
I don't know how the MobileVit structure or code is structured. Please refer to "usage/export/object_detection".