Dobby is !free/Intern 🐣

인턴일지🥕 분석용 이미지 만들기

컴공생 C 2021. 7. 28. 16:20
반응형

얻고 싶었던 것은 아래 이미지

결과화면

목표 : test_ma(우리모델)는 맞고 test_xlm(baseline)은 틀린 문제를 비교/ 분석해보기

문제의 구조를 살펴보면 img0, img1 이 있고, json에 "sent" 로 문장이 주어져있다.

img0, img1에 대해 문장이 참이면 True, 거짓이면 False를 반환하면 된다. 

처음에는 이미지 링크를 통해 가져와서 합치려고 했는데 일부 이미지가 저작권때문인지 접근자체가 되지 않아서 포기하고 이미 다운받아놓은 이미지 중 가능한 것들로 골라서 사용하기로 했다.

#필요한 라이브러리 import
from PIL import Image
import matplotlib.pyplot as plt
from matplotlib.image import imread

fig=plt.figure()
#이미지만 보이게 하기 위해
plt.axis('off')

#문제의 id, 정답, 문제내용을 제목으로 추가
plt.suptitle(str(idt)+" ("+ans+") "+'\n'+sent)
                
#PIL을 이용해 이미지 열기
left=Image.open(left_path)
right=Image.open(right_path)

#subplot 왼쪽 (1,2) 구조의 1번
ax1=fig.add_subplot(121)
ax1.axis('off')
ax1.imshow(left)

#subplot오른쪽 (1,2) 구조의 2번
ax2=fig.add_subplot(122)
ax2.axis('off')
ax2.imshow(right)
                
dstName="../analysis/"+str(idt)+".jpg"
                
plt.tight_layout()
#plt.show()

#plt 저장
plt.savefig(dstName,dpi=fig.dpi)

img0을 left로, img1을 right로 지정했다.

이게 뭐라고 거의 반나절을,, 끙끙

변명이라면 jupyter notebook에서 하려니 cv2.imshow() 이런게 안되어서 확인이 안되고 이미지 합치기(hstack)을 사용할 수 없었다는 점..?

일단 이걸 급하게 처리할 필요는 없어졌지만 고생한것이 의미있어,,기록,,

반응형