mlreco.post_processing.store.store_uresnet_ppn module

mlreco.post_processing.store.store_uresnet_ppn.store_uresnet_ppn(cfg, data_blob, res, logdir, iteration, nms_score_threshold=0.8, window_size=3, score_threshold=0.9, type_threshold=2, **kwargs)[source]
Configuration
  • input_data (str, optional)

  • store_method (str, optional)

  • threshold, size (NMS parameters)

  • score_threshold (to filter based on score only (no NMS))

mlreco.post_processing.store.store_uresnet_ppn.nms_numpy(im_proposals, im_scores, threshold, size)[source]