Actually, there are three places required to be modified listed as follows:
1、maskrcnn_benchmark/structures/segmentation_mask.py
2、maskrcnn_benchmark/modeling/balanced_positive_negative_sampler.py
3、maskrcnn-benchmark-master/maskrcnn_benchmark/modeling/rpn/inference.py
subscribe torch.bool with torch.uint8(上面三个地方只要出现torch.bool就改成torch.uint8)
And the most import thing is that you need to remove the build files
and execute ‘python setup.py build_ext install‘ again.(重构一下)
RuntimeError: empty is not implemented for type CUDABoolType
最新推荐文章于 2024-01-19 10:56:04 发布