Skip to content

DXW Conference Service Team2 코디 추천 & 가상 피팅 서비스 오모입(오늘 모 입지?)

Notifications You must be signed in to change notification settings

git-hubsalt/outfittransformer

Repository files navigation

outfittransformer

environment setting

  • pip install -r requirement.txt

dataset

  • Google Drive에서 download (original: AIHub 'KFashion 이미지' 데이터)
  • outfittransformer/data 폴더 생성
  • compatibility_train.txt, compatibility_valid.txt, item_metadata.json 파일 해당 경로로 이동

train

  • model_arg.py에서 주요 hyperparameter 수정 가능
python cp_train.py --task cp --train_batch 64 --valid_batch 96 --n_epochs 5 --learning_rate 1e-5 --scheduler_step_size 1000

About

DXW Conference Service Team2 코디 추천 & 가상 피팅 서비스 오모입(오늘 모 입지?)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages