question about train my dataset？

YonghaoHe / LFFD-A-Light-and-Fast-Face-Detector-for-Edge-Devices

A light and fast one class detection framework for edge devices. We provide face detector, head detector, pedestrian detector, vehicle detector......

MIT License

1.32k stars 329 forks source link

question about train my dataset？ #25

Closed leemengxing closed 3 years ago

leemengxing commented 5 years ago

Thank you for your work. I would like to ask if you train your own data set. Can we count the face size in our data set and delete the layers that do not exist in our data set? For example, my data set has only 320 * 240 image sizes and certainly no loss branch 8.

YonghaoHe commented 5 years ago

@leemengxing Of cause you can adjust the network to fit your case, if you are familiar with the method. In your case, 320*240 does not need large scales.