DACON Monthly Computer Vision Anomaly Detection Algorithm Competition

Algorithm | CV | Classification | Anomaly detection | Macro f1 score

 

metric 기반의 few shot learning

2022.05.09 16:50 2,117 Views language

https://github.com/sicara/easy-few-shot-learning
불량 이미지가 많지 않아서, Few shot 기반으로 위 프로젝트를 참고해 실험을 해봤습니다. 제일 간단한 Episodic training을 택했습니다
각 클래스마다 불량들만 추출해서 CUB Train set으로 한 후 Resnet12 기반으로 1 query 나머지는 shot으로 모두 넣은 후 훈련 진행했습니다 (클래스마다 불량 갯수가 달라서 위와 같이 선정했습니다)
제 생각에는 엄청 좋은 결과가 안나온 점이 mvtec이 imagenet, cifar와 다른 도메인이다보니 잘 되는 클래스가 있는 반면 0.2 accuracy로 낮은 클래스도 존재했습니다
대회에 늦게 참여해서, 간단히 실행 정도만 공유 드립니다. 다른 의견 있으시면 자유롭게 남겨주시면 좋겠습니다

Code
Login Required
0 / 1000
steven
2022.05.11 20:04

few shot learning을 돌려보고 싶은데 아래 코드에서 에러가 나네요
 from easyfsl.datasets import MVTEC 

ingbeeeded
2022.05.12 11:40

CUB와 동일하게 정의했습니다!