diff --git a/fastdeploy/vision/detection/ppdet/ppdet_pybind.cc b/fastdeploy/vision/detection/ppdet/ppdet_pybind.cc index 800d656b37d..67f63eb23b2 100644 --- 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()); + pybind11::class_(m, "PaddleYOLOv8") + .def(pybind11::init()); + pybind11::class_(m, "RTMDet") .def(pybind11::init()); + pybind11::class_(m, "CascadeRCNN") .def(pybind11::init()); + pybind11::class_(m, "PSSDet") .def(pybind11::init()); + pybind11::class_(m, "RetinaNet") .def(pybind11::init()); + pybind11::class_(m, "PPYOLOESOD") .def(pybind11::init()); + pybind11::class_(m, "FCOS") .def(pybind11::init()); + pybind11::class_(m, "TTFNet") .def(pybind11::init()); + pybind11::class_(m, "TOOD") .def(pybind11::init()); + pybind11::class_(m, "GFL") .def(pybind11::init());