diff --git a/tests/yolo3_tetrapack_resize/yolo3_tetrapack_resize.cpp b/tests/yolo3_tetrapack_resize/yolo3_tetrapack_resize.cpp index c1e772a..5a4012c 100644 --- a/tests/yolo3_tetrapack_resize/yolo3_tetrapack_resize.cpp +++ b/tests/yolo3_tetrapack_resize/yolo3_tetrapack_resize.cpp @@ -89,7 +89,7 @@ const char *output_bins[3] = { int main() { // Network layout - tk::dnn::dataDim_t dim(1, 3, 544, 320, 1); + tk::dnn::dataDim_t dim(1, 3, 320, 544, 1); tk::dnn::Network net(dim); tk::dnn::Conv2d c0 (&net, 32, 3, 3, 1, 1, 1, 1, c0_bin, true);