Commit edfd5a77 authored by Koen van de Sande's avatar Koen van de Sande Committed by Francisco Massa
Browse files

Fix reference training script for Mask R-CNN for PyTorch 1.2 (during...

Fix reference training script for Mask R-CNN for PyTorch 1.2 (during evaluation after epoch, mask datatype became bool, pycocotools expects uint8) (#1413)
parent 49b01e3a
...@@ -109,7 +109,7 @@ class CocoEvaluator(object): ...@@ -109,7 +109,7 @@ class CocoEvaluator(object):
labels = prediction["labels"].tolist() labels = prediction["labels"].tolist()
rles = [ 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 mask in masks
] ]
for rle in rles: for rle in rles:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment