Python predictor.COCODemo() Examples

The following are 1 code examples of predictor.COCODemo(). You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may also want to check out all available functions/classes of the module predictor , or try the search function .
Example #1
Source File: tracker.py    From argoverse_baselinetracker with MIT License 5 votes vote down vote up 
def __init__(
        self,
        region_type,
        ground_level, 
        ground_removal_th,
        use_maskrcnn,
        city_name,
        min_point_num,
        dbscan_eps,
        calib_data,
        path_debug_output,
        ground_removal_method,
        dataset_name,
        path_rcnn_config="../mask_rcnn/models/e2e_mask_rcnn_R_50_FPN_1x.yaml",

    ):
        self.region_type = region_type
        self.ground_level = ground_level
        self.ground_removal_th = ground_removal_th
        self.use_maskrcnn = use_maskrcnn
        self.city_name = city_name
        self.min_point_num = min_point_num
        self.dbscan_eps = dbscan_eps
        self.calib_data = calib_data
        self.path_debug_output = path_debug_output
        self.ground_removal_method = ground_removal_method
        self.dataset_name = dataset_name

        if use_maskrcnn:
            cfg.merge_from_file(path_rcnn_config)
            cfg.freeze()

            # using setting in official demo
            coco_demo_detector = COCODemo(
                cfg,
                confidence_threshold=0.7,
                min_image_size=224,
            )

            self.mask_rcnn_detector = coco_demo_detector