search

hustvl / BMaskR-CNN

[ECCV 2020] Boundary-preserving Mask R-CNN
https://arxiv.org/abs/2007.08921
Apache License 2.0
194 stars 41 forks source link

KeyError: "No object named 'BoundaryROIHeads' found in 'ROI_HEADS' registry!" #6

Closed PytaichukBohdan closed 4 years ago

PytaichukBohdan commented 4 years ago

When trying to run the demo.py, keep getting registry related problems. Have tried to modify the code in the detectron2 module, but unsuccessfully.

Is there anyone who encountered this problem? How did you solve it?

PytaichukBohdan commented 4 years ago

Or is there any other way to get alpha channel prediction from one image input? Please let me know if there is one.

wondervictor commented 4 years ago

Hi，you need to import bmaskrcnn in demo.py. Or directly copy the demo.py to projects/BMask-RCNN and import the bmaskrcnn module, then you can use the demo to obtain predictions

PytaichukBohdan commented 4 years ago

@wondervictor could you be so kind to provide with minimal working example? Have been struggling to make it work on 1 custom image for the last 3 days

charlescho64 commented 4 years ago

Hello, PytaichukBohdan

Are you done with this issue?

If done, how to get the result?

wondervictor commented 4 years ago

@charlescho64 @PytaichukBohdan Hi all, you need to import files/modules of BMask R-CNN to the demo.py to register the modules first.

charlescho64 commented 4 years ago

@wondervictor Thank you. I solved it. I am interesting in your working.

other, could you give "cascade_bmask_rcnn_R_101_FPN_1x.yaml" to me because there is weight file but not config file in your site?

wondervictor commented 4 years ago

@charlescho64 config file for R101 has been updated cascade_bmask_rcnn_R_101_FPN_1x.yaml