craft 학습결과 canvas문제

Question

안녕하세요  epoch 1000 에서 100까지만 돌려보고 결과확인하려고 solution코드를 돌려서 결과를 다운받았는데 여러 데이터가 저장되었는데 한가지사진으로 다 저장이 되었습니다.    # draw one sample training image for sanity check 이 부분에 canvas로 imshow되는 사진 하나만 저장이 되는데 예측결과 여러개 사진을 받아보려고 하면 어디 부분을 수정해야할까요?

AISchool · Answer

안녕하세요~. 반갑습니다. 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 · Answer

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