Skip to content

Commit

Permalink
[YOLOv8] add PaddleYOLOv8 pybind11 (#1144)
Browse files Browse the repository at this point in the history
* [Model] Support PaddleYOLOv8 model

* [YOLOv8] Add PaddleYOLOv8 pybind

* [Other] update from latest develop (#30)

* [Backend] Remove all lite options in RuntimeOption (#1109)

* Remove all lite options in RuntimeOption

* Fix code error

* move pybind

* Fix build error

* [Backend] Add TensorRT  FP16 support for AdaptivePool2d (#1116)

* add fp16 cuda kernel

* fix code bug

* update code

* [Doc] Fix KunlunXin doc (#1139)

fix kunlunxin doc

* [Model] Support PaddleYOLOv8 model (#1136)

Co-authored-by: Jason <jiangjiajun@baidu.com>
Co-authored-by: yeliang2258 <30516196+yeliang2258@users.noreply.github.com>

Co-authored-by: Jason <jiangjiajun@baidu.com>
Co-authored-by: yeliang2258 <30516196+yeliang2258@users.noreply.github.com>
  • Loading branch information
3 people authored Jan 14, 2023
1 parent 4e2102a commit f88c662
Showing 1 changed file with 12 additions and 0 deletions.
12 changes: 12 additions & 0 deletions fastdeploy/vision/detection/ppdet/ppdet_pybind.cc
Original file line number Diff line number Diff line change
Expand Up @@ -134,30 +134,42 @@ void BindPPDet(pybind11::module& m) {
.def(pybind11::init<std::string, std::string, std::string, RuntimeOption,
ModelFormat>());

pybind11::class_<vision::detection::PaddleYOLOv8, vision::detection::PPDetBase>(m, "PaddleYOLOv8")
.def(pybind11::init<std::string, std::string, std::string, RuntimeOption,
ModelFormat>());

pybind11::class_<vision::detection::RTMDet, vision::detection::PPDetBase>(m, "RTMDet")
.def(pybind11::init<std::string, std::string, std::string, RuntimeOption,
ModelFormat>());

pybind11::class_<vision::detection::CascadeRCNN, vision::detection::PPDetBase>(m, "CascadeRCNN")
.def(pybind11::init<std::string, std::string, std::string, RuntimeOption,
ModelFormat>());

pybind11::class_<vision::detection::PSSDet, vision::detection::PPDetBase>(m, "PSSDet")
.def(pybind11::init<std::string, std::string, std::string, RuntimeOption,
ModelFormat>());

pybind11::class_<vision::detection::RetinaNet, vision::detection::PPDetBase>(m, "RetinaNet")
.def(pybind11::init<std::string, std::string, std::string, RuntimeOption,
ModelFormat>());

pybind11::class_<vision::detection::PPYOLOESOD, vision::detection::PPDetBase>(m, "PPYOLOESOD")
.def(pybind11::init<std::string, std::string, std::string, RuntimeOption,
ModelFormat>());

pybind11::class_<vision::detection::FCOS, vision::detection::PPDetBase>(m, "FCOS")
.def(pybind11::init<std::string, std::string, std::string, RuntimeOption,
ModelFormat>());

pybind11::class_<vision::detection::TTFNet, vision::detection::PPDetBase>(m, "TTFNet")
.def(pybind11::init<std::string, std::string, std::string, RuntimeOption,
ModelFormat>());

pybind11::class_<vision::detection::TOOD, vision::detection::PPDetBase>(m, "TOOD")
.def(pybind11::init<std::string, std::string, std::string, RuntimeOption,
ModelFormat>());

pybind11::class_<vision::detection::GFL, vision::detection::PPDetBase>(m, "GFL")
.def(pybind11::init<std::string, std::string, std::string, RuntimeOption,
ModelFormat>());
Expand Down

0 comments on commit f88c662

Please sign in to comment.