v/GFPGAN
1
0
Fork 0
mirror of https://github.com/TencentARC/GFPGAN.git synced 2025-05-16 07:10:24 -07:00
Code Issues Projects Releases Wiki Activity

fix basicsr losses import (#210)

Browse Source
This commit is contained in:
AJ 2022-07-13 03:01:06 +01:00 committed by GitHub
parent c7ccc098a7
commit 7d657f26b6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
gfpgan/models/gfpgan_model.py
View File

@ -3,7 +3,7 @@ import os.path as osp
import torch
from basicsr.archs import build_network
from basicsr.losses import build_loss
from basicsr.losses.losses import r1_penalty
from basicsr.losses.gan_loss import r1_penalty
from basicsr.metrics import calculate_metric
from basicsr.models.base_model import BaseModel
from basicsr.utils import get_root_logger, imwrite, tensor2img