작성
·
187
0
Faster R-CNN을 이용한 Person Detection 실행 중
detections, predictions_dict, shapes = detect_fn(input_tensor) label_id_offset = 1 image_np_with_detections = image_np.copy() # image_np_with_detections np.array에 시각화 결과를 반환합니다. viz_utils.visualize_boxes_and_labels_on_image_array( image_np_with_detections, detections['detection_boxes'][0].numpy(), (detections['detection_classes'][0].numpy() + label_id_offset).astype(int), detections['detection_scores'][0].numpy(), category_index, use_normalized_coordinates=True, max_boxes_to_draw=200, min_score_thresh=.30, agnostic_mode=False ) plt.figure(figsize=(20,20)) plt.imshow(image_np_with_detections) plt.show()
이 셀에서
detections, predictions_dict, shapes = detect_fn(input_tensor)
이 부분에 오류 발생하네요.
detections, predictions_dict, shapes = detect_fn(input_tensor)
답변 1
0
cuda 라이브러리 초기화 문제 이슈가 있다고 합니다.
tensorflow와 cuda라이브러리를 아래와 같이 재 설치 후 해결되었습니다.
factory reset runtime in colab runtime
!pip install tensorflow==2.8
!apt install --allow-change-held-packages libcudnn8=8.1.0.77-1+cuda11.2
tensorflow 2.92에서
!apt install --allow-change-held-packages libcudnn8=8.1.0.77-1+cuda11.2
이것만 실행해도 잘 실행됩니다.