@@ -1,7 +1,7 @@
#include "infer/infer.hpp"
#include "infer/trt/yolo/yolo.hpp"
-#include "infer/trt/depth/yolo.hpp"
+#include "infer/trt/depth/depth.hpp"
#include <iostream>
#include <memory>
