바로 해볼 수 있는 딥러닝 사이트 Hugging face

이번엔 학습된 인공지능 모델을 테스트 해볼 수 있는 사이트를 알려드리려고 합니다.

 

먼저 Hugging face란?

트랜스포머 모델과 학습 스크립트를 제공하는 모듈입니다.

허깅 페이스를 사용하면, layer,model등을 선언하거나 학습 스크립트를 구현해야하는 수고를 덜 수 있습니다. 

 

그렇다면 트랜스포머(Transformer)란?

인공신경망 알고리즘은, 합성공(CNN), 순환(RNN) 신경망, 트랜스포머(Transformer)처럼 3가지로 나뉩니다.

트랜스포머는 구글이 2017년에 발표한 "Attention is all you need"에서 나온 모델입니다.

문장 전체를 병렬구조로 번역하고, 단어끼리의 연관성을 만들어 유사성을 높여 RNN의 한계를 극복한 모델이라고 합니다.

 

소스코드는 다음 처럼 간단하게 사용가능합니다.

from diffusers import StableDiffusionPipeline
import torch
model_id = "prompthero/openjourney"
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
pipe = pipe.to("cuda")
prompt = "retro serie of different cars with different colors and shapes, mdjrny-v4 style"
image = pipe(prompt).images[0]
image.save("./retro_cars.png")

 

https://huggingface.co/prompthero/openjourney?text=rainy+day 

prompthero/openjourney · Hugging Face

위에 작성한 예제는 미드저니처럼 나오게 하는 text2image 학습 모델입니다.

실제 미드저니에 비하면 퀄리티가 조금 떨어지긴 하지만 그래도 몇몇은 비슷하게 나옵니다.

그리고 오른쪽에 테스트를 해볼 수 있도록 되어 있습니다. 하지만 없는 경우도 있으니 참고하시면 될 것 같습니다.

딥러닝 쪽을 보고 있는데 예전에 했을 때에 비해 많은 것이 생긴것 같습니다.

(너무 못 따라가고 있는듯..)