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