search

ziyannchen / VFRxBenchmark

[NTIRE2024] official code for "Towards Real-world Video Face Restoration: A New Benchmark"
https://ziyannchen.github.io/projects/VFRxBenchmark/
Other
17 stars 0 forks source link

Model downloding issue when computing SER_FIQ #3

Open zycheiheihei opened 1 month ago

zycheiheihei commented 1 month ago

I got an ImportError when I try to compute the metric of SER-FIQ.

Traceback (most recent call last): File "scripts/inference_with_metrics.py", line 102, in main(root_path) File "scripts/inference_with_metrics.py", line 86, in main metric_results[target_metric].append(calculate_metric(dict(data=eval_data), metric_opt)) File "/opt/local-path-provisioner/zhangyichi/video-face-restoration/VFRxBenchmark/bfrxlib/metrics/init.py", line 28, in calculate_metric metric = METRIC_REGISTRY.get(metric_type)(data, opt) File "/opt/local-path-provisioner/zhangyichi/video-face-restoration/VFRxBenchmark/bfrxlib/metrics/serfiq_metric.py", line 25, in calculate_serfiq model = SER_FIQ(gpu=0 if device == 'cuda' else None) File "/opt/local-path-provisioner/zhangyichi/video-face-restoration/VFRxBenchmark/bfrxlib/metrics/serfiq/face_image_quality.py", line 63, in init from bfrxlib.utils.misc import uncompress_file ImportError: cannot import name 'uncompress_file' from 'bfrxlib.utils.misc' (/VFRxBenchmark/bfrxlib/utils/misc.py)

ziyannchen commented 1 month ago

Hello, there's a problem in this downloading script currently. I recommend to download the model file mannually here, then uncompress it within bfrxlib/metrics/serfiq/insightface/model/.

zycheiheihei commented 1 month ago

Thank you! However, serfiq/insightface/mtcnn-model/det1-0001.params for MTCNN is also missing. Can you provide another download link?

ziyannchen commented 1 month ago

SER-FIQ includes the weights in their repo, see this source. Download directly of all the .params and .caffemodel files (with a prefix from det1 to det4) in the serfiq/insightface/mtcnn-model/.