1580158
/
PBRemTools
mirror of https://github.com/mattyamonaca/PBRemTools.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #55 from d8ahazard/main 

Use shared models path
main
mattya_monaca 2024-11-04 15:05:07 +09:00 committed by GitHub
parent ff25bbf3ee 7741d8c08a
commit d38bfc3c97
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
scripts/tools.py
View File

@ -21,8 +21,13 @@ from collections import OrderedDict
from PIL import Image from PIL import Image
model_cache = OrderedDict() model_cache = OrderedDict()
sam_model_dir = os.path.join( models_path = shared.models_path
extensions_dir, "PBRemTools/models/") sams_dir = os.path.join(models_path, "sam")
if os.path.exists(sams_dir):
sam_model_dir = sams_dir
else:
sam_model_dir = os.path.join(
extensions_dir, "PBRemTools/models/")
model_list = [f for f in os.listdir(sam_model_dir) if os.path.isfile( model_list = [f for f in os.listdir(sam_model_dir) if os.path.isfile(
os.path.join(sam_model_dir, f)) and f.split('.')[-1] != 'txt'] os.path.join(sam_model_dir, f)) and f.split('.')[-1] != 'txt']