Beispiel #1
0
 def metrics(self):
     _, _, H, W = self.ishape
     metric = COCODetectionMetric(self.val_dataset,
                                  '_eval',
                                  cleanup=True,
                                  data_shape=(H, W))
     metric.reset()
     return metric
Beispiel #2
0
    def metrics(self):
        """ Customized metrics method introduction.

            COCODetectionMetric is used which is the detection metric for COCO bbox task.
        """
        _, _, H, W = self.ishape
        metric = COCODetectionMetric(
            self.val_dataset, '_eval', cleanup=True, data_shape=(H, W))
        metric.reset()
        return metric