인프런 커뮤니티 질문&답변
작성한 질문수
Region proposal network(RPN)
작성
·
389
0
강사님, Faster RCNN 의 Tensorflow code에서 RPN과 관련된 코드는 어디에 있는지 궁금합니다.
답변 2
1
권 철민
지식공유자
안녕하십니까,
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
