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
Merge request reports
Activity
Filter activity
Please register or sign in to reply