-
질문 & 답변
카테고리
-
세부 분야
컴퓨터 비전
-
해결 여부
미해결
craft 학습결과 canvas문제
21.06.09 19:13 작성 조회수 151
0
안녕하세요
epoch 1000 에서 100까지만 돌려보고 결과확인하려고 solution코드를 돌려서 결과를 다운받았는데 여러 데이터가 저장되었는데 한가지사진으로 다 저장이 되었습니다.
# draw one sample training image for sanity check
이 부분에 canvas로 imshow되는 사진 하나만 저장이 되는데 예측결과 여러개 사진을 받아보려고 하면 어디 부분을 수정해야할까요?
차량 번호판 인식 프로젝트와 TensorFlow로 배우는 딥러닝 영상인식 올인원
Custom Dataset 실전 프로젝트 실습 1 - CRAFT를 이용한 License Plate Detection 모델(Custom Dataset) 학습 Solution
강의실 바로가기
답변을 작성해보세요.
0
AISchool
지식공유자2021.06.10
안녕하세요~. 반갑습니다.
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
답변 2