snicolau commited on
Commit
a86ce83
Β·
verified Β·
1 Parent(s): 220b81a

Update lama/saicinpainting/evaluation/__init__.py

Browse files
lama/saicinpainting/evaluation/__init__.py CHANGED
@@ -2,8 +2,8 @@ import logging
2
 
3
  import torch
4
 
5
- from saicinpainting.evaluation.evaluator import InpaintingEvaluatorOnline, ssim_fid100_f1, lpips_fid100_f1
6
- from saicinpainting.evaluation.losses.base_loss import SSIMScore, LPIPSScore, FIDScore
7
 
8
 
9
  def make_evaluator(kind='default', ssim=True, lpips=True, fid=True, integral_kind=None, **kwargs):
 
2
 
3
  import torch
4
 
5
+ from lama.saicinpainting.evaluation.evaluator import InpaintingEvaluatorOnline, ssim_fid100_f1, lpips_fid100_f1
6
+ from lama.saicinpainting.evaluation.losses.base_loss import SSIMScore, LPIPSScore, FIDScore
7
 
8
 
9
  def make_evaluator(kind='default', ssim=True, lpips=True, fid=True, integral_kind=None, **kwargs):