2024年3月14日
flops&params
摘要： 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 阅读(13) 评论(0) 推荐(0)
计算lpips
摘要： 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 阅读(375) 评论(0) 推荐(0)