Advertisement

maskrcnn_benchmark学习内容基于Python实现的推理流程

阅读量:
复制代码
 主要看下run_on_opencv_image(self, image)。大概是这样几步。逐步记录

    
     predictions = self.compute_prediction(image)          #np.ndarray): an image as returned by OpenCV to tensor
    
     top_predictions = self.select_top_predictions(predictions)
    
     result = image.copy()
    
     result = self.overlay_boxes(result, top_predictions)
    
     if self.cfg.MODEL.KEYPOINT_ON:
    
     result = self.overlay_keypoints(result, top_predictions)
    
     result = self.overlay_class_names(result, top_predictions)
    
     return result
    
    
    
    

**compute_prediction(self, o

全部评论 (0)

还没有任何评论哟~