Inflearn Community Q&A
asked
craft 학습결과 canvas문제
Written on
·
309
0
안녕하세요
epoch 1000 에서 100까지만 돌려보고 결과확인하려고 solution코드를 돌려서 결과를 다운받았는데 여러 데이터가 저장되었는데 한가지사진으로 다 저장이 되었습니다.
# draw one sample training image for sanity check
이 부분에 canvas로 imshow되는 사진 하나만 저장이 되는데 예측결과 여러개 사진을 받아보려고 하면 어디 부분을 수정해야할까요?
cnn머신러닝 배워볼래요? 딥러닝컴퓨터-비전NLPtensorflow
Answer 2
0
AISchool
Instructor
안녕하세요~. 반갑습니다.
import imageio
output_folder = 'craft_license_plate_validation_result'
for image_path, _, _ in validation:
image = keras_ocr.tools.read(image_path)
output_image_path = os.path.join(output_folder, image_path.split('/')[-1])
# detector prediction
pred_boxes = detector.detect(np.expand_dims(image, axis=0))
for each_pred in pred_boxes[0]:
left, top = each_pred[0]
right, bottom = each_pred[2]
canvas = cv2.rectangle(image, (left, top), (right, bottom), (0,255,0), 3)
imageio.imwrite(output_image_path, canvas)
print(output_image_path + ' saved!' )
위 solution 코드에서 validation 부분에 visualization 하시려는 파일들의 경로를 list로 지정해서 사용하세요. (validation=['a.jpg','b.jpg','c.jpg'])
위 예시처럼한다면 3개 element를 가진 tuple에서 1개 element를 가진 list로 변경하니 아래와 같이 코드를 변경하셔야겠죠.
for image_path in validation:
감사합니다.
0
