Update test_wholeimage_swap_multispecific.py

by adding :
import warnings
warnings.filterwarnings("ignore")

we can get rid of the numerous errors and warnings that appear while doing image swaps.
This commit is contained in:
Fibonacci134
2022-05-07 16:04:40 -04:00
committed by GitHub
parent dd1ecdd2a7
commit d6b2ec6049
+2 -1
View File
@@ -24,7 +24,8 @@ import torch.nn as nn
from util.norm import SpecificNorm
import glob
from parsing_model.model import BiSeNet
import warnings
warnings.filterwarnings("ignore")
def lcm(a, b): return abs(a * b) / fractions.gcd(a, b) if a and b else 0
transformer_Arcface = transforms.Compose([