profile
Trendy AI Developer

DDIM_sampler 코드분석

DDIMsampler코드 **DDIM : Training 과정 중 noising - denoising 과정에서 Non-Markov chain 기존 DDPM에선 denoising에서 베이지안룰 과 마르코프 체인을 이용해서 모든 스텝t를 사용했지만 DDIM에선 t를 위해 x0과 xt-1만 사용 ** 클래스 선언 보통 ddim_sampler = DDIMSampler(model) 같이 model만 넘겨주는 듯 또한 보통 (B H W C)의 이미지를 (B C H W)이미지로 변환 후 모델로 넘겨야함 주요 함수들 sample() ddim_sampler 객체 생성 후 와 같이 바로 sample 함수로 이미지를 뽑음 역할은 샘플링 과정을 수행하기 전 입력 인자 확인 & 스케줄을 생성함 ddim_sampling() 위의 s

2023년 3월 7일
·
0개의 댓글
·

canny2image 코드분석(수정)

ControlNet_canny2image 코드 분석 gradio를 기반으로 작성된 코드임 > from share import * import config > import cv2 import einops import gradio as gr import numpy as np import torch import random > from pytorchlightning import seedeverything from annotator.util import resize_image, HWC3 from annotator.canny import CannyDetector from cldm.model import createmodel, loadstate_dict from cldm.ddim_hacked import DDIMSampler cv2.

2023년 3월 5일
·
0개의 댓글
·