|
@@ -396,19 +396,19 @@ bool YoloModelImpl::load(const std::string &engine_file, ModelType model_type, f
|
|
|
return true;
|
|
|
}
|
|
|
|
|
|
-data::BoxArray YoloModelImpl::forward(const tensor::Image &image, int slice_width, int slice_height, float overlap_width_ratio, float overlap_height_ratio, void *stream = nullptr)
|
|
|
+data::BoxArray YoloModelImpl::forward(const tensor::Image &image, int slice_width, int slice_height, float overlap_width_ratio, float overlap_height_ratio, void *stream)
|
|
|
{
|
|
|
slice_->slice(image, slice_width, slice_height, overlap_width_ratio, overlap_height_ratio, stream);
|
|
|
return forwards(stream);
|
|
|
}
|
|
|
|
|
|
-data::BoxArray YoloModelImpl::forward(const tensor::Image &image, void *stream = nullptr)
|
|
|
+data::BoxArray YoloModelImpl::forward(const tensor::Image &image, void *stream)
|
|
|
{
|
|
|
slice_->autoSlice(image, stream);
|
|
|
return forwards(stream);
|
|
|
}
|
|
|
|
|
|
-data::BoxArray YoloModelImpl::forwards(void *stream = nullptr)
|
|
|
+data::BoxArray YoloModelImpl::forwards(void *stream)
|
|
|
{
|
|
|
int num_image = slice_->slice_num_h_ * slice_->slice_num_v_;
|
|
|
if (num_image == 0) return {};
|