How to train the model

[webfrank]

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]

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]

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]

if use caffe, you can refer to ssd-models