readme

.vscode/c_cpp_properties.json

@@ -4,6 +4,7 @@
             "name": "Linux",
             "includePath": [
                 "${workspaceFolder}/src/**",
+                "${workspaceFolder}/include/**",
                 "/usr/local/cuda/include",
                 "/usr/local/include",
                 "/usr/include/**",

README.md

+
+
+1. 编译：
+
+git clone https://gitee.com/waytous/deepinfer.git
+cd DeepInfer
+
+修改CMakeLists.txt中cuda、tensorrt、yaml、gflag和glog等库的路径
+
+mkdir build
+cmake ..
+make -j32
+
+
+2. 测试：
+修改需要测试任务的配置文件中的configRootPath和device，configRootPath为配置文件configs的绝对路径，device为运行gpu的id
+
+./main TaskMOTS image ../configs/tasks/mots/mots.yaml ../test/bsw.jpg ../test/mots_res.jpg
+
+./main TaskDetect image ../configs/tasks/thermal/thermal.yaml ../test/thermal2.jpg,../test/thermal2.jpg ../test/t_b1_res.jpg,../test/t_b2_res.jpg
+
+./main TaskFusion image ../configs/tasks/fusion/fusion.yaml ../test/fusion_rgb.png,../test/fusion_thermal.png ../test/fusion_res.jpg
+
+./main TaskDMS image ../configs/tasks/dms/dms.yaml ../test/dms.jpg ../test/dms_res.jpg
+
+./main TaskBSW image ../configs/tasks/bsw/bsw.yaml ../test/bsw.jpg ../test/bsw_res.jpg
+
+
+3. 扩展
+本库可以分为三级扩展：Unit、Model和Task
+3.1 Unit
+    继承BaseUnit基类，从GlobalIOMap中获取输入，最后将输出加入此map中
+    每个Unit的输入输出都是继承至BaseIO类，根据不同Unit的需要进行继承。
+
+3.2 Model
+    继承基类BaseModel，将输入的图像或者点云等资源进行注册，运行定义好的全部Unit，输出结果
+
+3.3 Task
+    继承基类BaseTask，根据不同的任务，将不同模型的输入输出结合并处理
+
+
+
+

configs/tasks/bsw/bsw.yaml

 
 configRootPath: /home/wangxin/projects/waytous/DeepInfer
-device: 2
+device: 0
 
 detectorName: CameraModel
 detectorConfigPath: configs/tasks/bsw/rgb_yolov5.yaml

configs/tasks/dms/dms.yaml

 
 configRootPath: /home/wangxin/projects/waytous/DeepInfer
-device: 2
+device: 0
 
 
 faceDetectorName: CameraModel

configs/tasks/fusion/fusion.yaml

 
 configRootPath: /home/wangxin/projects/waytous/DeepInfer
-device: 2
+device: 0
 
 modelName: CameraModel
 modelConfigPath: configs/tasks/fusion/fusion_model.yaml

configs/tasks/mots/mots.yaml

 
 configRootPath: /home/wangxin/projects/waytous/DeepInfer
-device: 2
+device: 0
 segmentorName: CameraModel
 segmentorConfigPath: configs/tasks/mots/trades.yaml
 

configs/tasks/thermal/thermal.yaml

 
 configRootPath: /home/wangxin/projects/waytous/DeepInfer
-device: 3
+device: 0
 detectorName: CameraModel
 detectorConfigPath: configs/tasks/thermal/yolov5.yaml