@@ -58,6 +58,9 @@ namespace depth
output_buffer_.gpu(input_numel / 3);
output_buffer_.cpu(input_numel / 3);
+
+ affine_matrix_.gpu(6);
+ affine_matrix_.cpu(6);
}
void preprocess(const tensor::Image &image, affine::LetterBoxMatrix &affine, void *stream = nullptr);