neuralet icon indicating copy to clipboard operation
neuralet copied to clipboard

Generate edge-TPU compiled TF-lite from Face Mask detector model for Coral

Open mhejrati opened this issue 4 years ago • 5 comments

mhejrati avatar May 18 '20 15:05 mhejrati

@mhejrati I generated tflite model and sent a PR for this task on neuralet-models repo.

mrn-mln avatar May 24 '20 15:05 mrn-mln

@mrn-mln Cool, thanks.

Can you please provide more details of what you have done there and what are the next steps?

mhejrati avatar Jun 03 '20 04:06 mhejrati

@mhejrati A ssd-mobilenet-v1 detector is trained on face-mask dataset. I converted the model to quantized int8 model which is compatible with edgetpu devices (the mAP = 0.67%). At the next step, I am going to use the tflite model for implementing a face/face-mask detector app at the social-distancing repo. The task is related to issue 106.

mrn-mln avatar Jun 03 '20 14:06 mrn-mln

I tested a ssd-movilenet-v1, the generalization of the model was not well. So, I will try to train a new model for this task. I will update my activity here.

mrn-mln avatar Jun 09 '20 13:06 mrn-mln

I trained resnet-50 on WIDER FACE dataset. The mAP was ~0.52 which was not well enough. I will try to train a model on the HollywoodHead detection dataset in order to detect head.

mrn-mln avatar Jun 12 '20 19:06 mrn-mln