Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Error in utils/segment/general
masks2segments()
(#9724)
When running segmentation predict on gpu, the conversion from tensor to numpy fails. Calling `.cpu()` solves this problem. Signed-off-by: Paul Guerrie <97041392+paulguerrie@users.noreply.github.com> Signed-off-by: Paul Guerrie <97041392+paulguerrie@users.noreply.github.com>
- Loading branch information