search

AIZOOTech / FaceMaskDetection

开源人脸口罩检测模型和数据 Detect faces and determine whether people are wearing mask.
https://demo.aizoo.com/face-mask-detection.html
MIT License
2.08k stars 618 forks source link

Pytorch version #51

Closed tempdata73 closed 4 years ago

tempdata73 commented 4 years ago

I was just wondering in what Pytorch version were the models trained on? I want to get rid of those pesky SourceChange warnings...

Thank you for releasing these models!

AIZOOTech commented 4 years ago

I trained this model with keras, then I convert the model to PyTorch version. You can load the model, and save it again. Maybe this can solve the warning message.

twmht commented 3 years ago

is pytorch really from keras?

I convert mxnet model (I believe this is also from keras) to pytorch, but with input 360x360, the number of the generated bounding boxes are always 2 for one face.