├── .gitignore ├── CITATION.cff ├── LICENSE ├── LICENSE-DATASET ├── README.md ├── bash_scripts ├── briaai.sh ├── compute_mask.sh ├── evaluation.sh ├── fine_tune_hyperparameters.sh ├── make_a_copy.sh ├── photoroom.sh ├── rm_cache.sh ├── split.sh ├── sub.sh ├── test.sh ├── train.sh ├── train_finetuning.sh ├── train_test.sh └── visualization.sh ├── birefnet ├── __init__.py ├── config.py ├── dataset.py ├── evaluation │ └── metrics.py ├── gen_best_ep.py ├── image_proc.py ├── loss.py ├── models │ ├── backbones │ │ ├── build_backbone.py │ │ ├── pvt_v2.py │ │ └── swin_v1.py │ ├── birefnet.py │ ├── modules │ │ ├── aspp.py │ │ ├── decoder_blocks.py │ │ ├── deform_conv.py │ │ ├── lateral_blocks.py │ │ ├── prompt_encoder.py │ │ └── utils.py │ └── refinement │ │ ├── refiner.py │ │ └── stem_layer.py └── utils.py ├── images ├── dataset_composition.png ├── dataset_example.jpg ├── models_comparison.png └── test_image_samurai.jpg ├── requirements.txt ├── scripts ├── briaai.py ├── compute_mask.py ├── eval_existingOnes.py ├── evaluations.py ├── inference.py ├── photoroom.py ├── split.py ├── train.py └── visualization.py ├── setup.py └── toonout_demo.ipynb /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/.gitignore -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/CITATION.cff -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/LICENSE -------------------------------------------------------------------------------- /LICENSE-DATASET: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/LICENSE-DATASET -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/README.md -------------------------------------------------------------------------------- /bash_scripts/briaai.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/bash_scripts/briaai.sh -------------------------------------------------------------------------------- /bash_scripts/compute_mask.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/bash_scripts/compute_mask.sh -------------------------------------------------------------------------------- /bash_scripts/evaluation.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/bash_scripts/evaluation.sh -------------------------------------------------------------------------------- /bash_scripts/fine_tune_hyperparameters.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/bash_scripts/fine_tune_hyperparameters.sh -------------------------------------------------------------------------------- /bash_scripts/make_a_copy.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/bash_scripts/make_a_copy.sh -------------------------------------------------------------------------------- /bash_scripts/photoroom.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/bash_scripts/photoroom.sh -------------------------------------------------------------------------------- /bash_scripts/rm_cache.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/bash_scripts/rm_cache.sh -------------------------------------------------------------------------------- /bash_scripts/split.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/bash_scripts/split.sh -------------------------------------------------------------------------------- /bash_scripts/sub.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/bash_scripts/sub.sh -------------------------------------------------------------------------------- /bash_scripts/test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/bash_scripts/test.sh -------------------------------------------------------------------------------- /bash_scripts/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/bash_scripts/train.sh -------------------------------------------------------------------------------- /bash_scripts/train_finetuning.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/bash_scripts/train_finetuning.sh -------------------------------------------------------------------------------- /bash_scripts/train_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/bash_scripts/train_test.sh -------------------------------------------------------------------------------- /bash_scripts/visualization.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/bash_scripts/visualization.sh -------------------------------------------------------------------------------- /birefnet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /birefnet/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/birefnet/config.py -------------------------------------------------------------------------------- /birefnet/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/birefnet/dataset.py -------------------------------------------------------------------------------- /birefnet/evaluation/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/birefnet/evaluation/metrics.py -------------------------------------------------------------------------------- /birefnet/gen_best_ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/birefnet/gen_best_ep.py -------------------------------------------------------------------------------- /birefnet/image_proc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/birefnet/image_proc.py -------------------------------------------------------------------------------- /birefnet/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/birefnet/loss.py -------------------------------------------------------------------------------- /birefnet/models/backbones/build_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/birefnet/models/backbones/build_backbone.py -------------------------------------------------------------------------------- /birefnet/models/backbones/pvt_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/birefnet/models/backbones/pvt_v2.py -------------------------------------------------------------------------------- /birefnet/models/backbones/swin_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/birefnet/models/backbones/swin_v1.py -------------------------------------------------------------------------------- /birefnet/models/birefnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/birefnet/models/birefnet.py -------------------------------------------------------------------------------- /birefnet/models/modules/aspp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/birefnet/models/modules/aspp.py -------------------------------------------------------------------------------- /birefnet/models/modules/decoder_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/birefnet/models/modules/decoder_blocks.py -------------------------------------------------------------------------------- /birefnet/models/modules/deform_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/birefnet/models/modules/deform_conv.py -------------------------------------------------------------------------------- /birefnet/models/modules/lateral_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/birefnet/models/modules/lateral_blocks.py -------------------------------------------------------------------------------- /birefnet/models/modules/prompt_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/birefnet/models/modules/prompt_encoder.py -------------------------------------------------------------------------------- /birefnet/models/modules/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/birefnet/models/modules/utils.py -------------------------------------------------------------------------------- /birefnet/models/refinement/refiner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/birefnet/models/refinement/refiner.py -------------------------------------------------------------------------------- /birefnet/models/refinement/stem_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/birefnet/models/refinement/stem_layer.py -------------------------------------------------------------------------------- /birefnet/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/birefnet/utils.py -------------------------------------------------------------------------------- /images/dataset_composition.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/images/dataset_composition.png -------------------------------------------------------------------------------- /images/dataset_example.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/images/dataset_example.jpg -------------------------------------------------------------------------------- /images/models_comparison.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/images/models_comparison.png -------------------------------------------------------------------------------- /images/test_image_samurai.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/images/test_image_samurai.jpg -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/briaai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/scripts/briaai.py -------------------------------------------------------------------------------- /scripts/compute_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/scripts/compute_mask.py -------------------------------------------------------------------------------- /scripts/eval_existingOnes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/scripts/eval_existingOnes.py -------------------------------------------------------------------------------- /scripts/evaluations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/scripts/evaluations.py -------------------------------------------------------------------------------- /scripts/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/scripts/inference.py -------------------------------------------------------------------------------- /scripts/photoroom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/scripts/photoroom.py -------------------------------------------------------------------------------- /scripts/split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/scripts/split.py -------------------------------------------------------------------------------- /scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/scripts/train.py -------------------------------------------------------------------------------- /scripts/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/scripts/visualization.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/setup.py -------------------------------------------------------------------------------- /toonout_demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MatteoKartoon/BiRefNet/HEAD/toonout_demo.ipynb --------------------------------------------------------------------------------