[11기HuggingFace] 제조공정시 볼트 불량품이미지 cnn으로 분류하기

KAMP 제조 데이터 셋

(Image Data) 정상 이미지 1,378개, 불량 이미지 74개, 데이터수1,452개

<데이터 수집>

중소벤처기업부, Korea AI Manufacturing Platform(KAMP),
제조AI데이터셋 제목, 제공기관(수행기관), 데이터셋등록일, KAMP URL

정상으로 구분되는 1102개

불량으로 구분되는 59개

테스트로 사용되는 이미지 (정상 276개, 불량 15개)

정상: 1102

불량: 59

정상: 276

불량: 15

print(X.shape, y.shape)

(1161, 128, 128, 3) (1161,)

학습데이터 링크

코랩 소스코드 링크

허깅페이스에서 불량품 분류하는것을 못찾아서, 직접 코랩으로 해보았습니다.

챗지피티로 cnn 모델 분석 소스코드 문의한 내용입니다.

vgg16을 모델을 불러와서 학습하기

#11기HuggingFace