Skip to content

스케치 이미지에서 500개의 클래스를 분류하는 프로젝트

Notifications You must be signed in to change notification settings

Heejin1002/Sketch_Image_Classification

 
 

Repository files navigation

모델

  • config : 경로, 모델명, 모델 파라미터 관련 내용
  • models : timm 모델 load
  • model_CLIP : hugging_face 모델 load
  • data_loader : 데이터셋 구축
  • model_selector : 모델 선택
  • transform : 이미지 증강
  • utils : 정확도
  • trainer : 모델 학습 및 검증
  • main : 종합

앙상블

  • ensoemble : 모델 로드, 앙상블 모델, 모델 학습
  • main_ensemble : 종합

파인튜닝

  • fineturning : CLIP_finetuning

생성 모델

  • GAN : 외부 이미지 증강

ipynbFile

  • ipynb로 제출하려고 하였습니다. 추후 github에는 ipynb로 제출하지 않은 편이 좋다고 알게 되어서 프로젝트 완료 후 추후 모듈화를 진행하였습니다. 모듈화에 대한 확신이 없어 위같이 ipynbfile도 추가로 제출하였습니다.

About

스케치 이미지에서 500개의 클래스를 분류하는 프로젝트

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 89.9%
  • Python 10.1%