完善文档

38a0aee1 · shizhm · aacfc4cd · 38a0aee1
Commit 38a0aee1 authored Jun 19, 2023 by shizhm
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

Src/YOLOV5.cpp Src/YOLOV5.cpp +1 -1

No files found.
--- a/Src/YOLOV5.cpp
+++ b/Src/YOLOV5.cpp
@@ -64,7 +64,7 @@ ErrorCode DetectorYOLOV5::Initialize(InitializationParameterOfDetector initializ
        }
        
        migraphx::onnx_options onnx_options;
-        onnx_options.map_input_dims["images"]={1,3,800,800};// 
+        onnx_options.map_input_dims["images"]={1,3,800,800};
        net = migraphx::parse_onnx(modelPath, onnx_options);
        LOG_INFO(stdout,"succeed to load model: %s\n",GetFileName(modelPath).c_str());