theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-24 10:18:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

add quality metric

Browse Source
This commit is contained in:
YunhongLu-ZJU
2025-02-14 13:59:56 +08:00
parent acda7d891a
commit a834371d50
3 changed files with 14 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
diffsynth/extensions/QualityMetric/config.py
View File

@@ -1,11 +1,15 @@
import os import os
CURRENT_DIR = os.path.dirname(os.path.abspath(__file__)) current_dir = os.path.dirname(os.path.abspath(__file__))
project_root = os.path.abspath(os.path.join(current_dir, '../../../'))
quality_metric_path = os.path.join(project_root, 'models', 'QualityMetric')
# 打印生成的路径
print(quality_metric_path)
def get_model_path(model_name): def get_model_path(model_name):
return os.path.join(CURRENT_DIR, MODEL_FOLDER, model_name) return os.path.join(quality_metric_path, model_name)
MODEL_FOLDER = "reward_pretrained"
MODEL_PATHS = { MODEL_PATHS = {
"aesthetic_predictor": get_model_path("aesthetic-predictor/sac+logos+ava1-l14-linearMSE.safetensors"), "aesthetic_predictor": get_model_path("aesthetic-predictor/sac+logos+ava1-l14-linearMSE.safetensors"),

17
examples/QualityMetric/README.md
View File

@@ -14,17 +14,12 @@ The file directory is shown below.
``` ```
DiffSynth-Studio/ DiffSynth-Studio/
└── diffsynth/ └── models/
└── extensions/ └── QualityMetric/
└── QualityMetric/ ├── HPS_v2/
├── __init__.py │ ├── HPS_v2_compressed.safetensors
├── hps.py │ ├── HPS_v2.1_compressed.safetensors
├── reward_pretrained/ └── ...
│ ├── HPS_v2/
│ │ ├── HPS_v2_compressed.safetensors
│ │ ├── HPS_v2.1_compressed.safetensors
│ └── ...
└── ...
``` ```
### Step 2: Test image quality metric ### Step 2: Test image quality metric

1
models/QualityMetric/Put pretrained reward checkpoints here.txt Normal file
View File

@@ -0,0 +1 @@