摘要: 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 阅读全文
posted @ 2024-03-14 15:24 helloWorldhelloWorld 阅读(3) 评论(0) 推荐(0) 编辑
摘要: import os import torch import lpips from PIL import Image from torchvision.transforms import ToTensor # 初始化LPIPS模型 loss_fn = lpips.LPIPS(net='vgg') de 阅读全文
posted @ 2024-03-14 11:43 helloWorldhelloWorld 阅读(36) 评论(0) 推荐(0) 编辑