Skip to content

Conversation

shigabeev
Copy link

@shigabeev shigabeev commented Aug 1, 2024

Changes:

  • Changed deepface to RetinaFace face detector. Why? Pytorch, GPU support.
  • Added Noise and grain detection filter. I have to refactor model weights somehow though.
  • Added Grayscale detection filter. It detects if image is grayscale even if it has 3 dimensions and was compressed.
  • Added Oily skin detection filter. Useful for portrait images.

Feel free to leave comments for refactoring. If you want a faster communication channel, ping me in telegram.

Usage:

 focus_filter = FaceFocusFilter(threshold=2000.0, detect_face=True, workers=1, batch_size=1) 
noise_filter = NoiseEstimationFilter(model_path='noise_estimator_model.joblib', params_path='feature_params.joblib', workers=4, batch_size=4) 
grayscale_filter = GrayscaleFilter(workers=1, batch_size=1) 
oily_skin_filter = OilySkinFilter(workers=1, batch_size=16, model_name="dima806/skin_types_image_detection") 
@shigabeev shigabeev changed the title New filters New filters: Noise, grayscale and oily skin detection Aug 1, 2024
@shigabeev shigabeev mentioned this pull request Dec 18, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

1 participant