Add NanoDet-Plus Model support (#32)

* 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

* add yolox c++/pybind and model_zoo demo

* move some helpers to private

* fixed CMakeLists.txt typos

* add normalize with alpha and beta

* add version notes for yolov5/yolov6/yolox

* add copyright to yolov5.cc

* revert normalize

* fixed some bugs in yolox

* Add NanoDet-Plus Model support

Co-authored-by: Jason <jiangjiajun@baidu.com>
This commit is contained in:
DefTruth
2022-07-22 09:49:55 +08:00
committed by GitHub
parent a8458e6729
commit e248781784
16 changed files with 907 additions and 7 deletions

View File

@@ -23,6 +23,7 @@ void BindPPSeg(pybind11::module& m);
void BindUltralytics(pybind11::module& m);
void BindMeituan(pybind11::module& m);
void BindMegvii(pybind11::module& m);
void BindRangiLyu(pybind11::module& m);
#ifdef ENABLE_VISION_VISUALIZE
void BindVisualize(pybind11::module& m);
#endif
@@ -56,6 +57,7 @@ void BindVision(pybind11::module& m) {
BindWongkinyiu(m);
BindMeituan(m);
BindMegvii(m);
BindRangiLyu(m);
#ifdef ENABLE_VISION_VISUALIZE
BindVisualize(m);
#endif