摘要:
from thop import profile import time if __name__ == '__main__': net = LLFormer() net = net.to(torch.device('cuda:0')) input = torch.randn(1, 3, 256, 2 阅读全文
摘要:
import os import torch import lpips from PIL import Image from torchvision.transforms import ToTensor # 初始化LPIPS模型 loss_fn = lpips.LPIPS(net='vgg') de 阅读全文