How to train the model - Githubissues

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

How to train the model #37

Open webfrank opened 4 years ago

webfrank commented 4 years ago

Hi, I'm just a novice and I would like to know if you can share the train code (for any model or pytorch) to setup a complete pipeline.

One more question is what are the differences between face_mask_detection and model360 ?

AIZOOTech commented 4 years ago

Hi, this model is a simple SSD framework. Training process is quite easy. You can find many open source SSD implementation on Github. The model360 is the face mask detection with 360x360 input size, the other is 260x260 input size. Hope everything is fine.

simonbuehler commented 4 years ago

hi,

thanks for your work, we collected and annotated some difficult face_mask images and i'd like to retrain the network with those new images. could you publish your train script, i guess it should be easy to export the model afterwards with onnx for other frameworks?

imistyrain commented 3 years ago

if use caffe, you can refer to ssd-models