Skip to content

Releases: LanceHuang245/ImageGuard

ImageGuard is now available!

Choose a tag to compare

@LanceHuang245 LanceHuang245 released this 18 Feb 04:26

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