Skip to content

Releases: LanceHuang245/ImageGuard

ImageGuard is now available!

18 Feb 04:26

Choose a tag to compare

ImageGuard

  • Based on efficientnet-b3
  • Predictive image classification (neutral, porn)
  • The model was only trained for two epochs for experimental purposes only (about 78% accuracy)

How to use?

  • Create model folder in project
  • Copy image_guard_v2.pth into the model folder
  • Start flask service