mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-06 17:17:14 +08:00

Files

DefTruth de7c06a309 add yolov6 c++ and yolov6 pybind (#16 )

* update .gitignore

* Added checking for cmake include dir

* fixed missing trt_backend option bug when init from trt

* remove un-need data layout and add pre-check for dtype

* changed RGB2BRG to BGR2RGB in ppcls model

* add model_zoo yolov6 c++/python demo

* fixed CMakeLists.txt typos

* update yolov6 cpp/README.md

2022-07-14 16:12:28 +08:00

CMakeLists.txt

add yolov6 c++ and yolov6 pybind (#16 )

2022-07-14 16:12:28 +08:00

README.md

add yolov6 c++ and yolov6 pybind (#16 )

2022-07-14 16:12:28 +08:00

yolov6.cc

add yolov6 c++ and yolov6 pybind (#16 )

2022-07-14 16:12:28 +08:00

README.md

编译YOLOv6示例

# 下载和解压预测库
wget https://bj.bcebos.com/paddle2onnx/fastdeploy/fastdeploy-linux-x64-0.0.3.tgz
tar xvf fastdeploy-linux-x64-0.0.3.tgz

# 编译示例代码
mkdir build & cd build
cmake ..
make -j

# 下载模型和图片
wget https://github.com/meituan/YOLOv6/releases/download/0.1.0/yolov6s.onnx
wget https://raw.githubusercontent.com/ultralytics/yolov5/master/data/images/bus.jpg

# 执行
./yolov6_demo

执行完后可视化的结果保存在本地vis_result.jpg，同时会将检测框输出在终端，如下所示

DetectionResult: [xmin, ymin, xmax, ymax, score, label_id]
11.772949,229.269287, 792.933838, 748.294189, 0.954794, 5
667.140381,396.185455, 807.701721, 881.810120, 0.900997, 0
223.271011,405.105743, 345.740723, 859.328552, 0.898938, 0
50.135777,405.863129, 245.485519, 904.153809, 0.888936, 0
0.000000,549.002869, 77.864723, 869.455017, 0.614145, 0