add watermark

.gitignore

@@ -1,3 +1,5 @@
 __pycache__
 hivision/creator/weights/*
-!hivision/creator/weights/.gitkeep
+!hivision/creator/weights/.gitkeep
+hivision/creator/retinaface/weights/*
+!hivision/creator/retinaface/weights/.gitkeep

README.md

@@ -1,12 +1,12 @@
 # HivisionIDPhotos-ComfyUI
-a custom node for [HivisionIDPhotos](https://github.com/Zeyi-Lin/HivisionIDPhotos), you can find [workflow](./doc/证件照_workflow.json)
+a custom node for [HivisionIDPhotos](https://github.com/Zeyi-Lin/HivisionIDPhotos), you can find [证件照_workflow](./doc/证件照_workflow.json),or [id_photo_workflow](./doc/id_photo_workflow.json)
 
-![comfyui demo](doc/demo_2.png)
+![comfyui demo](doc/web.png)
 
 ## Example
-| 输入 | 抠图 | 换背景 | 排版 |
-| -- | -- | -- | -- |
-| ![](./doc/demo.jpg) [source](https://www.liblib.art/imageinfo/b7cb6b18b2af4c37be8607b648b52979) | ![](./doc/ComfyUI_temp_movvp_00002_.png) ![](./doc/ComfyUI_temp_igzmq_00002_.png) | ![](./doc/ComfyUI_temp_byppo_00004_.png) | ![](./doc/ComfyUI_temp_jeppc_00005_.png) |
+| 输入 | 抠图 | 换背景 | 加水印 | 排版 |
+| -- | -- | -- | -- | -- |
+| ![](./doc/demo.jpg) [source](https://www.liblib.art/imageinfo/b7cb6b18b2af4c37be8607b648b52979) | ![](./doc/ComfyUI_temp_movvp_00002_.png) ![](./doc/ComfyUI_temp_igzmq_00002_.png) | ![](./doc/ComfyUI_temp_byppo_00004_.png) | ![](./doc/ComfyUI_temp_yhlxo_00002_.png)|![](./doc/ComfyUI_temp_jeppc_00005_.png) |
 
 ## 教程
 - [Demo](https://www.bilibili.com/video/BV1iFpvegEY3/)
@@ -16,5 +16,7 @@ a custom node for [HivisionIDPhotos](https://github.com/Zeyi-Lin/HivisionIDPhoto
 存到项目的`ComfyUI/custom_nodes/HivisionIDPhotos-ComfyUI/hivision/creator/weights`目录下：
 - `modnet_photographic_portrait_matting.onnx` (24.7MB): [MODNet](https://github.com/ZHKKKe/MODNet)官方权重，[下载](https://github.com/Zeyi-Lin/HivisionIDPhotos/releases/download/pretrained-model/modnet_photographic_portrait_matting.onnx)
 - `hivision_modnet.onnx` (24.7MB): 对纯色换底适配性更好的抠图模型，[下载](https://github.com/Zeyi-Lin/HivisionIDPhotos/releases/download/pretrained-model/hivision_modnet.onnx)
-- `mnn_hivision_modnet.mnn` (24.7MB): mnn转换后的抠图模型 by [zjkhahah](https://github.com/zjkhahah)，[下载](https://github.com/Zeyi-Lin/HivisionIDPhotos/releases/download/pretrained-model/mnn_hivision_modnet.mnn)
-- `rmbg-1.4.onnx` (176.2MB): [BRIA AI](https://huggingface.co/briaai/RMBG-1.4) 开源的抠图模型，[下载](https://huggingface.co/briaai/RMBG-1.4/resolve/main/model.pth?download=true)后重命名为`rmbg-1.4.onnx`
+- `rmbg-1.4.onnx` (176.2MB): [BRIA AI](https://huggingface.co/briaai/RMBG-1.4) 开源的抠图模型，[下载](https://huggingface.co/briaai/RMBG-1.4/resolve/main/onnx/model.onnx?download=true)后重命名为`rmbg-1.4.onnx`
+- `birefnet-v1-lite.onnx`(224MB): [ZhengPeng7](https://github.com/ZhengPeng7/BiRefNet) 开源的抠图模型，[下载](https://github.com/ZhengPeng7/BiRefNet/releases/download/v1/BiRefNet-general-bb_swin_v1_tiny-epoch_232.onnx)后重命名为`birefnet-v1-lite.onnx`
+
+- RetinaFace | **离线**人脸检测模型，CPU推理速度中等（秒级），精度较高| [下载](https://github.com/Zeyi-Lin/HivisionIDPhotos/releases/download/pretrained-model/retinaface-resnet50.onnx)后放到`hivision/creator/retinaface/weights`目录下