|
|
@@ -335,7 +335,7 @@ void YoloModelImpl::adjust_memory(int batch_size)
|
|
|
box_count_.cpu(1);
|
|
|
}
|
|
|
|
|
|
-void YoloModelImpl::preprocess(int ibatch, affine::LetterBoxMatrix &affine, void *stream = nullptr)
|
|
|
+void YoloModelImpl::preprocess(int ibatch, affine::LetterBoxMatrix &affine, void *stream)
|
|
|
{
|
|
|
affine.compute(std::make_tuple(slice_->slice_width_, slice_->slice_height_),
|
|
|
std::make_tuple(network_input_width_, network_input_height_));
|
