Wallpaper defect types Classification AI Competition

Algorithm | Vision | Classification | MLOps | Weighted F1 Score

  • moneyIcon Prize : 10,000,000 KRW
  • 2,116 Users Completed

 

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

2023.05.31 09:17 1,986 Views language

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

Code
Login Required
0 / 1000
plmokn
2023.06.08 09:39

멋진 코드네요