Use setBackend instead of ConvertToCpu workaround
Change the convertion to backend 'cpu' from a custom function converToCpu
in the database to the generic setBackend()
.
Content :
-
Remove the call to convertToCpu in database, its definition & tests.
However it raises a problem on the layout of the data between different backends.
In the current code the resulting tensor with backend 'cpu' have layout [N,H,W,C] instead of [N,C,H,W].
-
Use format HWC for OpenCV tensor and change format when converting to CPU backend.
Edited by Olivier BICHLER