Support loading model weights from local project folders

Add bgfilter/weights.py: resolve_model_source() maps a HuggingFace repo id
to a local folder under the weights dir (models/ by default, override with
BGFILTER_WEIGHTS_DIR) when one named after the repo basename exists; otherwise
the repo id is returned unchanged. Opt-in and backward compatible.

- vitmatte_infer.py / segmentation.py resolve model_name through it; anime-seg
  reads <dir>/isnetis.onnx directly instead of hf_hub_download when local.
- .gitignore: models/, model-cache/
- README: document bundling weights as plain project folders.

Verified: anime-seg loads from a local models/anime-seg/ folder offline.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>

This commit is contained in:

孔令弘

2026-07-03 02:50:19 +00:00

parent 7a87a00ca1

commit da4164e95a

5 changed files with 51 additions and 4 deletions

.gitignore

View File

@@ -1,4 +1,6 @@
 Samples/
 Outputs/
 models/
 model-cache/
 __pycache__/
 *.py[cod]