leon 1 bulan lalu
induk
melakukan
dd0c62a2dc
1 mengubah file dengan 3 tambahan dan 3 penghapusan
  1. 3 3
      src/infer/trt/yolo.cu

+ 3 - 3
src/infer/trt/yolo.cu

@@ -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 {};