apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-05 16:48:03 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Move cpp code to directory csrcs (#42)

Browse Source 
* move cpp code to csrcs

* move cpp code to csrcs
This commit is contained in:
Jason
2022-07-26 17:59:02 +08:00
committed by GitHub
parent 7fa3afa9de
commit ffbc5cc42d
128 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

76
fastdeploy/vision/vision_pybind.cc
View File

@@ -1,76 +0,0 @@
// Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#include "fastdeploy/pybind/main.h"
namespace fastdeploy {
void BindPPCls(pybind11::module& m);
void BindPPDet(pybind11::module& m);
void BindWongkinyiu(pybind11::module& m);
void BindPPSeg(pybind11::module& m);
void BindUltralytics(pybind11::module& m);
void BindMeituan(pybind11::module& m);
void BindMegvii(pybind11::module& m);
void BindDeepCam(pybind11::module& m);
void BindRangiLyu(pybind11::module& m);
#ifdef ENABLE_VISION_VISUALIZE
void BindVisualize(pybind11::module& m);
#endif
void BindVision(pybind11::module& m) {
pybind11::class_<vision::ClassifyResult>(m, "ClassifyResult")
.def(pybind11::init())
.def_readwrite("label_ids", &vision::ClassifyResult::label_ids)
.def_readwrite("scores", &vision::ClassifyResult::scores)
.def("__repr__", &vision::ClassifyResult::Str)
.def("__str__", &vision::ClassifyResult::Str);
pybind11::class_<vision::DetectionResult>(m, "DetectionResult")
.def(pybind11::init())
.def_readwrite("boxes", &vision::DetectionResult::boxes)
.def_readwrite("scores", &vision::DetectionResult::scores)
.def_readwrite("label_ids", &vision::DetectionResult::label_ids)
.def("__repr__", &vision::DetectionResult::Str)
.def("__str__", &vision::DetectionResult::Str);
pybind11::class_<vision::FaceDetectionResult>(m, "FaceDetectionResult")
.def(pybind11::init())
.def_readwrite("boxes", &vision::FaceDetectionResult::boxes)
.def_readwrite("scores", &vision::FaceDetectionResult::scores)
.def_readwrite("landmarks", &vision::FaceDetectionResult::landmarks)
.def_readwrite("landmarks_per_face",
&vision::FaceDetectionResult::landmarks_per_face)
.def("__repr__", &vision::FaceDetectionResult::Str)
.def("__str__", &vision::FaceDetectionResult::Str);
pybind11::class_<vision::SegmentationResult>(m, "SegmentationResult")
.def(pybind11::init())
.def_readwrite("masks", &vision::SegmentationResult::masks)
.def("__repr__", &vision::SegmentationResult::Str)
.def("__str__", &vision::SegmentationResult::Str);
BindPPCls(m);
BindPPDet(m);
BindPPSeg(m);
BindUltralytics(m);
BindWongkinyiu(m);
BindMeituan(m);
BindMegvii(m);
BindDeepCam(m);
BindRangiLyu(m);
#ifdef ENABLE_VISION_VISUALIZE
BindVisualize(m);
#endif
}
} // namespace fastdeploy