CS/์ธ๊ณต์ง๋ฅ
2024. 6. 24.
[Tensorflow keras] image generation using Stable Diffusion
๋ชฉํ TensorFlow Keras library ๋ฅผ ์ฌ์ฉํด์ CIFAR-10 dataset ๊ฐ์ ์ด๋ฏธ์ง๋ฅผ ์์ฑํ๋ ๊ฐ๋จํ diffusion ๋ชจ๋ธ์ ๋ง๋ค์ด๋ณผ ๊ฒ์ด๋ค.๋ค์์ diffusion ๋ชจ๋ธ์ด ๋์ํ๋ ๋ฐฉ์์ด๊ณ , fully noised ์ด๋ฏธ์ง์์๋ถํฐ ์ ์ง์ ์ผ๋ก ์ด๋ฏธ์ง๋ฅผ ์์ฑํด๋๊ฐ๋ค. ์์์ ๋ดค๋ฏ์ด, ์ด ๋ชจ๋ธ์ ์ ์ง์ ์ผ๋ก ๋
ธ์ด์ฆ๋ฅผ ์ค์ฌ๊ฐ๋ฉด์ ์ด๋ฏธ์ง๋ฅผ ์์ฑํ๋ค. ๋ฐ๋ผ์ ๋ณธ ๋ฅ๋ฌ๋ ๋ชจ๋ธ์๊ฒ noise ๋ฅผ ์ค์ด๋ ๋ฐฉ๋ฒ์ ํ์ต์ํฌ ๊ฒ์ด๋ฉฐ ์ด๋ฅผ ์ํด ๋๊ฐ์ง ์
๋ ฅ์ด ํ์ํ๋ค. - ์
๋ ฅ : ์ฒ๋ฆฌ๋์ด์ผํ ์ด๊ธฐ Noise - timestamp : noise status Importing necessary libraries import numpy as npfrom tqdm.auto import trange, ..