Mask_R-CNN_demo assert error

🐛 Bug

Error when run the webcam or notebook demo

---------------------------------------------------------------------------
AssertionError                            Traceback (most recent call last)
<ipython-input-16-ba06eae97bcc> in <module>()
      1 # compute predictions
----> 2 predictions = coco_demo.run_on_opencv_image(image)
      3 imshow(predictions)

/Users/an.tran/Desktop/map-workspace/code/maskrcnn-benchmark/demo/predictor.pyc in run_on_opencv_image(self, image)
    167                 the BoxList via `prediction.fields()`
    168         """
--> 169         predictions = self.compute_prediction(image)
    170         top_predictions = self.select_top_predictions(predictions)
    171 

/Users/an.tran/Desktop/map-workspace/code/maskrcnn-benchmark/demo/predictor.pyc in compute_prediction(self, original_image)
    212             # in the image, as defined by the bounding boxes
    213             masks = prediction.get_field("mask")
--> 214             masks = self.masker(masks, prediction)
    215             prediction.add_field("mask", masks)
    216         return prediction

/Users/an.tran/Desktop/map-workspace/code/maskrcnn-benchmark/maskrcnn_benchmark/modeling/roi_heads/mask_head/inference.pyc in __call__(self, masks, boxes)
    183 
    184         # Make some sanity check
--> 185         assert len(boxes) == len(masks), "Masks and boxes should have the same length."
    186 
    187         # TODO:  Is this JIT compatible?

AssertionError: Masks and boxes should have the same length.

To Reproduce

Steps to reproduce the behavior:

1. On Mac
2. After installation, run the Python notebook Mask_R-CNN_demo.ipynb

Expected behavior

Environment

Please copy and paste the output from the environment collection script from PyTorch (or fill out the checklist below manually).

You can get the script and run it with:

wget https://raw.githubusercontent.com/pytorch/pytorch/master/torch/utils/collect_env.py
# For security purposes, please check the contents of collect_env.py before running it.
python collect_env.py

Collecting environment information… PyTorch version: 1.0.0.dev20181119 Is debug build: No CUDA used to build PyTorch: None

OS: Mac OSX 10.13.6 GCC version: Could not collect CMake version: Could not collect

Python version: 2.7 Is CUDA available: No CUDA runtime version: No CUDA GPU models and configuration: No CUDA Nvidia driver version: No CUDA cuDNN version: No CUDA

Versions of relevant libraries: [pip] 18.0 [conda] Could not collect

Additional context

[antran89]

@fmassa it solved!

[fmassa]

I’ve fixed this issue in #187 Let me know if you still face the same issue, and thanks for reporting it!