apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

apply lint

This commit is contained in:
Sean Sube 2023-02-12 13:32:50 -06:00
parent 62fec58f78
commit ae8aa2ea7a
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 33 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
api/onnx_web/hacks.py
View File

@ -47,17 +47,39 @@ def unload(exclude):
# these should be the same sources and names as `convert.base_models.sources`, but inverted so the source is the key # these should be the same sources and names as `convert.base_models.sources`, but inverted so the source is the key
cache_path_map = { cache_path_map = {
"https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/codeformer.pth": "correction-codeformer.pth", "https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/codeformer.pth": (
"https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/detection_Resnet50_Final.pth": "detection-resnet50-final.pth", "correction-codeformer.pth"
"https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/detection_mobilenet0.25_Final.pth": "detection-mobilenet025-final.pth", ),
"https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/parsing_bisenet.pth": "parsing-bisenet.pth", "https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/detection_Resnet50_Final.pth": (
"https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/parsing_parsenet.pth": "parsing-parsenet.pth", "detection-resnet50-final.pth"
"https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/RealESRGAN_x2plus.pth": "upscaling-real-esrgan-x2-plus", ),
"https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/yolov5l-face.pth": "detection-yolo-v5-l.pth", "https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/detection_mobilenet0.25_Final.pth": (
"https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/yolov5n-face.pth": "detection-yolo-v5-n.pth", "detection-mobilenet025-final.pth"
"https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth": "correct-gfpgan-v1-3.pth", ),
"https://s3.eu-central-1.wasabisys.com/nextml-model-data/codeformer/weights/facelib/detection_Resnet50_Final.pth": "detection-resnet50-final.pth", "https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/parsing_bisenet.pth": (
"https://s3.eu-central-1.wasabisys.com/nextml-model-data/codeformer/weights/facelib/parsing_parsenet.pth": "parsing-parsenet.pth", "parsing-bisenet.pth"
),
"https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/parsing_parsenet.pth": (
"parsing-parsenet.pth"
),
"https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/RealESRGAN_x2plus.pth": (
"upscaling-real-esrgan-x2-plus"
),
"https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/yolov5l-face.pth": (
"detection-yolo-v5-l.pth"
),
"https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/yolov5n-face.pth": (
"detection-yolo-v5-n.pth"
),
"https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth": (
"correct-gfpgan-v1-3.pth"
),
"https://s3.eu-central-1.wasabisys.com/nextml-model-data/codeformer/weights/facelib/detection_Resnet50_Final.pth": (
"detection-resnet50-final.pth"
),
"https://s3.eu-central-1.wasabisys.com/nextml-model-data/codeformer/weights/facelib/parsing_parsenet.pth": (
"parsing-parsenet.pth"
),
} }