How to train on new dataset

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 on new dataset #47

Open tiwari93 opened 4 years ago

tiwari93 commented 4 years ago

Hi, First of all thanks for doing wonderful job.

I am new to this field but want to create for my society with custom images. Please help in training the model and use that for detection

maaz-bin-asad commented 4 years ago

Hello , This is a problem on object detection. Since you are working with supervised learning, You will need a labelled dataset that contains images along with the object marked (In this case, mask) . Prefer using YOLO algorithm for this . Once your model gets trained ,it will output the probability that a person is wearing mask or not.

ZIKO94ZIKO commented 4 years ago

@maaz-bin-asad , @tiwari93 Could you send me your model please?