[YOLOv8] add PaddleYOLOv8 pybind11 (#1144)

* [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 <[email protected]> Co-authored-by: yeliang2258 <[email protected]> Co-authored-by: Jason <[email protected]> Co-authored-by: yeliang2258 <[email protected]>
PaddlePaddle · Jan 14, 2023 · f88c662 · f88c662
1 parent 4e2102a
commit f88c662
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/fastdeploy/vision/detection/ppdet/ppdet_pybind.cc b/fastdeploy/vision/detection/ppdet/ppdet_pybind.cc
@@ -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>());