From 00a8177acdfb2d643ccc00d4099afae5eef53f02 Mon Sep 17 00:00:00 2001 From: Koen van de Sande Date: Fri, 4 Oct 2019 11:04:17 +0200 Subject: [PATCH] Fix reference training script for Mask R-CNN for PyTorch 1.2 (during evaluation after epoch, mask datatype became bool, pycocotools expects uint8) --- references/detection/coco_eval.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/references/detection/coco_eval.py b/references/detection/coco_eval.py index 3460fa66456..d758a64a909 100644 --- a/references/detection/coco_eval.py +++ b/references/detection/coco_eval.py @@ -109,7 +109,7 @@ def prepare_for_coco_segmentation(self, predictions): labels = prediction["labels"].tolist() rles = [ - mask_util.encode(np.array(mask[0, :, :, np.newaxis], order="F"))[0] + mask_util.encode(np.array(mask[0, :, :, np.newaxis], dtype=np.uint8, order="F"))[0] for mask in masks ] for rle in rles: