-
카테고리
-
세부 분야
컴퓨터 비전
-
해결 여부
미해결
Region proposal network(RPN)
20.10.17 22:16 작성 조회수 332
0
강사님, Faster RCNN 의 Tensorflow code에서 RPN과 관련된 코드는 어디에 있는지 궁금합니다.
답변을 작성해보세요.
1
권 철민
지식공유자2020.10.18
안녕하십니까,
RPN관련된 코드라 하심은 어떤것인지요? 아래와 같이 graph 모델에서 rpn tensor를 찾기를 원하시는 건지요? 만일 그렇다면 저도 찾아봤는데, tensor 모델이 너무 복잡해서 rpn에 해당하는 tensor를 찾지 못했습니다. 다만 구글링을 해보니 'map/TensorArrayStack/TensorArrayGatherV3:0' 에 해당하는 tensor라고 합니다만( https://stackoverflow.com/questions/61877482/figure-out-the-name-of-rpn-output-in-faster-r-cnn-tensorflow-graph ) 정확히 매핑하지는 못했습니다.
out = sess.run([sess.graph.get_tensor_by_name('num_detections:0'),
sess.graph.get_tensor_by_name('detection_scores:0'),
sess.graph.get_tensor_by_name('detection_boxes:0'),
sess.graph.get_tensor_by_name('detection_classes:0')],
feed_dict={'image_tensor:0': inp.reshape(1, inp.shape[0], inp.shape[1], 3)})
0
답변 2