dovidovision/koclip-train

PyTorch Train code for Korean CLIP

koclip-train

PyTorch Train code for Korean CLIP

koclip-train's Issues

label 목록입니다.

label_names.txt

1. 놀란 고양이
2. 화난 고양이
3. 슬픈 고양이
4. 기쁜 고양이
5. 졸린 고양이
6. 달리는 고양이
7. 노는 고양이
8. 앉아있는 고양이
9. 누워있는 고양이
10. 궁금한 고양이
11. 웃는 고양이
12. 짜증난 고양이
13. 양말신은 고양이
14. 노트북하는 고양이
15. 하품하는 고양이
16. 선글라스 낀 고양이
17. 표독스러운 고양이
18. 샤워하는 고양이
19. 친구만나는 고양이
20. 그루밍하는 고양이
21. 사냥하는 고양이
22. 밥먹는 고양이
23. 물 마시는 고양이
24. 츄르먹는 고양이
25. 집사랑 노는 고양이
26. 냥냥펀치를 하는 고양이
27. 박스 안에 있는 고양이

기존 koclip과의 성능 비교

벤치마크 데이터셋인 cifar10에서의 Zero-shot image classification 정확도 비교

• 기존 koclip의 정확도 : 12.5%

• Ours : 56.85%

• 우리의 모델은 Pytorch 모델을 기반으로 하여 GPU 로드하기도 쉬움.