 Paul Guerrie
|
5ef69ef3e6
|
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>
|
2022-10-06 22:55:15 +02:00 |
|