¯\_(ツ)_/¯ 팀의 [카카오 아레나] 쇼핑몰 상품 카테고리 분류 코드입니다.
- python 3.6
- tensorflow >= 1.12
config의 경로를 위치에 맞게 설정한다.
python train.py
저장 경로는 기본적으로 상위 폴더(../)이며 이 폴더에 "UmsoImg_v2_e00"와 같은 이름으로 저장된다.
python inference.py # default UmsoImg_v2_e31
python inference.py --path UmsoImg_v2_e15 # example
https://drive.google.com/file/d/1ltPPALttaP5s_R4_gVbO4ByOpl2-gzfi/view?usp=sharing
위 모델의 크기는 182MB 입니다.