support model General-Lite-2K and model Matting
This commit is contained in:
@@ -25,7 +25,9 @@ Support the use of new and old versions of BiRefNet models
|
||||
|
||||
- General: A pre-trained model for general use cases.
|
||||
- General-Lite: A light pre-trained model for general use cases.
|
||||
- General-Lite-2K: A light pre-trained model for general use cases in high resolution (2560x1440).
|
||||
- Portrait: A pre-trained model for human portraits.
|
||||
- Matting: A pre-trained model for general trimap-free matting use.
|
||||
- DIS: A pre-trained model for dichotomous image segmentation (DIS).
|
||||
- HRSOD: A pre-trained model for high-resolution salient object detection (HRSOD).
|
||||
- COD: A pre-trained model for concealed object detection (COD).
|
||||
@@ -36,7 +38,9 @@ Model files go here (when use AutoDownloadBiRefNetModel automatically downloaded
|
||||
If necessary, they can be downloaded from:
|
||||
- [General](https://huggingface.co/ZhengPeng7/BiRefNet/resolve/main/model.safetensors) ➔ `model.safetensors` must be renamed `General.safetensors`
|
||||
- [General-Lite](https://huggingface.co/ZhengPeng7/BiRefNet_T/resolve/main/model.safetensors) ➔ `model.safetensors` must be renamed `General-Lite.safetensors`
|
||||
- [General-Lite-2K](https://huggingface.co/ZhengPeng7/BiRefNet_lite-2K/resolve/main/model.safetensors) ➔ `model.safetensors` must be renamed `General-Lite-2K.safetensors`
|
||||
- [Portrait](https://huggingface.co/ZhengPeng7/BiRefNet-portrait/resolve/main/model.safetensors) ➔ `model.safetensors` must be renamed `Portrait.safetensors`
|
||||
- [Matting](https://huggingface.co/ZhengPeng7/BiRefNet-matting/resolve/main/model.safetensors) ➔ `model.safetensors` must be renamed `Matting.safetensors`
|
||||
- [DIS](https://huggingface.co/ZhengPeng7/BiRefNet-DIS5K/resolve/main/model.safetensors) ➔ `model.safetensors` must be renamed `DIS.safetensors`
|
||||
- [HRSOD](https://huggingface.co/ZhengPeng7/BiRefNet-HRSOD/resolve/main/model.safetensors) ➔ `model.safetensors` must be renamed `HRSOD.safetensors`
|
||||
- [COD](https://huggingface.co/ZhengPeng7/BiRefNet-COD/resolve/main/model.safetensors) ➔ `model.safetensors` must be renamed `COD.safetensors`
|
||||
@@ -50,8 +54,8 @@ Some models on GitHub:
|
||||
- [BiRefNet-ep480.pth](https://huggingface.co/ViperYX/BiRefNet/resolve/main/BiRefNet-ep480.pth)
|
||||
|
||||
## Weight Models (Optional)
|
||||
- [swin_large_patch4_window12_384_22kto1k.pth](https://huggingface.co/ViperYX/BiRefNet/resolve/main/swin_large_patch4_window12_384_22kto1k.pth)(not General-Lite model)
|
||||
- [swin_tiny_patch4_window7_224_22kto1k_finetune.pth](https://drive.google.com/drive/folders/1cmce_emsS8A5ha5XT2c_CZiJzlLM81ms)(just General-Lite model)
|
||||
- [swin_large_patch4_window12_384_22kto1k.pth](https://huggingface.co/ViperYX/BiRefNet/resolve/main/swin_large_patch4_window12_384_22kto1k.pth)(not General-Lite and General-Lite-2K model)
|
||||
- [swin_tiny_patch4_window7_224_22kto1k_finetune.pth](https://drive.google.com/drive/folders/1cmce_emsS8A5ha5XT2c_CZiJzlLM81ms)(just General-Lite and General-Lite-2K model)
|
||||
|
||||
|
||||
## Nodes
|
||||
|
||||
Reference in New Issue
Block a user