我想预测其中一个类别,即“人”,而不是对所有被检查和预测的84个类别都进行预测。关于YOLACT的参考资料请参阅https://github.com/dbolya/yolact。预测结果还不错,但我认为我只需要修改其中一段代码,且修改方式应该很简单,但我找不到方法。
在此问题中,我按照他提到的方法执行了相关操作,如在Yolact/layers/output_utils.py中添加了4行代码,但没有进行其他更改。这四行代码如下:
在此问题中,我按照他提到的方法执行了相关操作,如在Yolact/layers/output_utils.py中添加了4行代码,但没有进行其他更改。这四行代码如下:
boxes = torch.cat((boxes[classes==0], boxes[classes==2]),dim=0)
scores = torch.cat((scores[classes==0], scores[classes==2]),dim=0)
masks = torch.cat((masks[classes==0], masks[classes==2]),dim=0)
classes = torch.cat((classes[classes==0], classes[classes==2]),dim=0)
但是它会产生以下错误:
RuntimeError: strides[cur - 1] == sizes[cur] * strides[cur] INTERNAL ASSERT FAILED at
/opt/conda/conda-bld/pytorch_1573049310284/work/torch/csrc/jit/fuser/executor.cpp:175,
please report a bug to PyTorch.
The above operation failed in interpreter, with the following stack trace:
terminate called without an active exception
Aborted (core dumped)
我尝试按照提到的方法添加if条件,但仍然出错。我正在使用PyTorch 1.3。