craft 학습결과 canvas문제

차량 번호판 인식 프로젝트와 TensorFlow로 배우는 딥러닝 영상인식 올인원

Custom Dataset 실전 프로젝트 실습 1 - CRAFT를 이용한 License Plate Detection 모델(Custom Dataset) 학습 Solution

작성

247

안녕하세요

epoch 1000 에서 100까지만 돌려보고 결과확인하려고 solution코드를 돌려서 결과를 다운받았는데 여러 데이터가 저장되었는데 한가지사진으로 다 저장이 되었습니다.

# draw one sample training image for sanity check

이 부분에 canvas로 imshow되는 사진 하나만 저장이 되는데 예측결과 여러개 사진을 받아보려고 하면 어디 부분을 수정해야할까요?

cnn 머신러닝 배워볼래요? 딥러닝 컴퓨터 비전 NLP tensorflow

답변 2

AISchool

지식공유자

안녕하세요~. 반갑습니다.

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:
감사합니다.

actyun1216

질문자

사진이 보시는거와같이 하나만으로 저장됩니다. 저는 학습된 사진 전부를 보고싶은데..

인프런 커뮤니티 질문&답변

craft 학습결과 canvas문제