도배 하자 유형 분류 AI 경진대회

알고리즘 | 비전 | 분류 | MLOps | Weighted F1 Score

  • moneyIcon 상금 : 1,000 만원
  • 2,059명 마감

 

[public 30, private 51] vit_base_resnet50 timm 라이브러리 사용

공동작성자

stroke
2023.05.31 09:17 908 조회 language

데이터 증강 후 로보플로우로 캐글이든, 코랩이든 쓸수있게 터미널환경에서 데이터셋 구축후 작업 하였습니다.
다소 허접하지만 timm 라이브러리로 모델을 찾다가 vit 베이스의 resnet50 의 모델이 있길래 이거다 싶어서 해봤는데 스코어는 잘 나왔지만
아무래도 전처리 싸움이 아니었나 생각합니다.

코드
로그인이 필요합니다
0 / 1000
plmokn
2023.06.08 09:39

멋진 코드네요