Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

RC-4.5.0: AutoInterface introduce #118

Merged
merged 24 commits into from
Feb 8, 2023
Merged

RC-4.5.0: AutoInterface introduce #118

merged 24 commits into from
Feb 8, 2023

Conversation

OPHoperHPO
Copy link
Owner

@OPHoperHPO OPHoperHPO commented Jan 3, 2023

Changelog:

  1. Removed obsolete models
  2. Fixed typos in tracer b7
  3. Added new AutoScene preprocessing method
  4. Added warning about fp16 to u2net class
  5. Updated README.
  6. Changelog
  7. Added ISNet segmentation network Support isnet.pth? (Dichotomous Image Segmentation) #119
  8. Added object classifier YOLOv4.
  9. Added AutoInterface to automatically select the best methods for images with different scene types.
    AutoInterface uses a scene classifier and an object classifier to perform full analysis on all images. At the moment, since there are not enough models for so many types of scenes, universal models are selected for some domains. In the future, when some variety of models is added, auto-selection will be rewritten for the better.
  10. Updated metrics
  11. Added CascadePSP segmentation refinement network.
    It will refine the segmentation mask before passing it to the matting network. This should provide more accurate background removal for objects and people. (Check repo from CREDITS.md for more details)
  12. Fixed trimap generator unknown area value as FBA Matting expect.
  13. Added noise filter switch (needed for smooth mask prediction on the final stage for soft scenes)
  14. Updated alpha composition algorithm

AutoScene RC-4.2.0

Description

It performs a primary analysis of the image in order to automatically select the necessary method for removing the background.
The choice is made by classifying the scene type.

Model accuracy:

Model achieves 91.3% accuracy on the validation set.

Classes info

The output can be the following types:

  • hard
  • soft
  • digital*

The hard class denotes a group of scenes to which a coarser background removal method should be applied, intended for objects with an edge without small details.
The hard class contains the following categories of objects: object, laptop, charger, pc mouse, pc, rocks, table, bed, box, sneakers, ship, wire, guitar, fork, spoon, plate, keyboard, car, bus, screwdriver, ball, door, flower, clocks, fruit , food, robot.

The soft class denotes a group of scenes to which you want to apply a soft background removal method intended for people, hair, clothes, and other similar types of objects.
The soft class contains the following categories of objects: animal, people, human, man, woman, t-shirt, hairs, hair, dog, cat, monkey, cow, medusa, clothes

The digital* class denotes a group of images with digital graphics, such as screenshots, logos, and so on.
The digital class contains the following categories of scenes: screenshot

* - Not fully supported at this moment

More details on model here:  https://huggingface.co/Carve/scene_classifier/

Architecture

image
The classifier uses DenseNet161 as the encoder and some linear layers at classifier base.

Some examples

image 1
image 1 (3)

Fixed typos in tracer b7
prepared for adding postprocessing and preprocessing
2. Added warning about fp16 to u2net class

AutoScene Description:
        Performs a primary analysis of the image in order to automatically select the necessary method for removing the background.
        The choice is made by classifying the scene type.

        The output can be the following types:
        - hard
        - soft
        - digital

        More info here:  https://huggingface.co/Carve/scene_classifier/
…erfaces.

2. Updated Russian README.md
3. Updated examples
@OPHoperHPO OPHoperHPO added the feature Feature label Jan 3, 2023
@OPHoperHPO OPHoperHPO self-assigned this Jan 3, 2023
MrFox131 and others added 18 commits January 5, 2023 15:36
…setup not being fully configured. Fixed models download(added scene classifier) and tests(same here).
Added AutoInterface to automatically select the best methods for images with different scene types. AutoInterface uses a scene classifier and an object classifier to perform full analysis on all images. At the moment, since there are not enough models for so many types of scenes, universal models are selected for some domains. In the future, when some variety of models is added, auto-selection will be rewritten for the better.
…ods for images with different scene types. auto uses a scene classifier and an object classifier to perform full analysis on all images. At the moment, since there are not enough models for so many types of scenes, universal models are selected for some domains. In the future, when some variety of models is added, auto-selection will be rewritten for the better.
It will refine the segmentation mask before passing it to the matting network. This should provide more accurate background removal for objects and people. (Check repo from CREDITS.md for more details)
Updated README.md
Updated Russian README.md
Optimized FBA input image size
Changelog
1.  Added ISNet segmentation network
2. Added object classifier YOLOv4.
3. Added AutoInterface to automatically select the best methods for images with different scene types.
AutoInterface uses a scene classifier and an object classifier to perform full analysis on all images. At the moment, since there are not enough models for so many types of scenes, universal models are selected for some domains. In the future, when some variety of models is added, auto-selection will be rewritten for the better.
4.  Updated metrics
5. Added CascadePSP segmentation refinement network.
It will refine the segmentation mask before passing it to the matting network. This should provide more accurate background removal for objects and people. (Check repo from CREDITS.md for more details)
6. Fixed trimap generator `unknown` area value as FBA Matting expect.  
7. Added noise filter switch (needed for smooth mask prediction on the final stage)
8. Updated alpha composition algorithm
@OPHoperHPO OPHoperHPO changed the title RC-4.2.0: AutoScene introduce RC-4.5.0: AutoInterface introduce Feb 8, 2023
@OPHoperHPO OPHoperHPO merged commit d89fa88 into development Feb 8, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
feature Feature
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants