/
githubmirror
/
stablediffusion
Обзор
Документация
Войти
/
githubmirror
/
stablediffusion
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
ldm/data/util.py
24 строки
629 B
Robin Rombach
release more models
24 ноя 2022, 03:22
24 ноя 2022, 03:22
ca86da3
Код
Авторство
О чём код?
import torch from ldm.modules.midas.api import load_midas_transform class AddMiDaS(object): def __init__(self, model_type): super().__init__() self.transform = load_midas_transform(model_type) def pt2np(self, x): x = ((x + 1.0) * .5).detach().cpu().numpy() return x def np2pt(self, x): x = torch.from_numpy(x) * 2 - 1. return x def __call__(self, sample): # sample['jpg'] is tensor hwc in [-1, 1] at this point x = self.pt2np(sample['jpg']) x = self.transform({"image": x})["image"] sample['midas_in'] = x return sample