Sfoglia il codice sorgente

Merge pull request #447 from FlyingQianMM/develop_new

fix numpy version
Jason 5 anni fa
parent
commit
23bf1df81e
1 ha cambiato i file con 2 aggiunte e 0 eliminazioni
  1. 2 0
      paddlex/cv/models/utils/detection_eval.py

+ 2 - 0
paddlex/cv/models/utils/detection_eval.py

@@ -996,6 +996,7 @@ def coco_error_analysis(eval_details_file=None,
             makeplot(recThrs, ps[:, :, k], res_out_dir, nm['name'], iou_type)
         makeplot(recThrs, ps, res_out_dir, 'allclass', iou_type)
 
+    np.linspace = fixed_linspace
     coco_gt = COCO()
     coco_gt.dataset = gt
     coco_gt.createIndex()
@@ -1006,4 +1007,5 @@ def coco_error_analysis(eval_details_file=None,
     if pred_mask is not None:
         coco_dt = loadRes(coco_gt, pred_mask)
         _analyze_results(coco_gt, coco_dt, res_type='segm', out_dir=save_dir)
+    np.linspace = backup_linspace
     logging.info("The analysis figures are saved in {}".format(save_dir))