Open f-lab-sage opened 8 months ago
DB 부분은 superbase를 고려해보기
PR 작성 방법
config.yaml
env:
env_name: dev, stage,
db:
endpoint: xxx
timeout:xxx
model:
batch_size:
model_path: s3://xxxxx
실제 로딩할 때는 load_yaml(xxx)
from PIL import Image
# init 과정
vton = VTryOnModel(start_epoch=1, epoch_iter=0)
vton.load_model()
seg = ClothSegmentationModel()
# 유저 요청이 들어왔을 때
person_img = Image.open("xxx.jpg")
cloth_img = Image.open("yyy.jpg")
dataset, dataset_size = vton.load_data("데이터들 저장된 경로")
# segmentation
# 전처리
pp_image = vton.preprocess("이미지 경로들")
output_img = vton.inference(pp_image)
output_img.save("")
# upload
예시