GithubStar/ncnn
1
0
Fork 0
mirror of https://github.com/Tencent/ncnn.git synced 2026-10-23 00:59:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

add pytest net (#2593)

Browse Source 
* add pytest net

* fix typo error

* fix header year
This commit is contained in:
Cai Shanli
2021-01-15 15:41:17 +08:00
committed by GitHub
parent 4f2907d129
commit 2785398eca
4 changed files with 95 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/src/main.cpp
+2 -2
View File
@@ -883,8 +883,8 @@ PYBIND11_MODULE(ncnn, m)
.def("clear", &Net::clear)
.def("create_extractor", &Net::create_extractor)
.def("blobs", &Net::blobs)
.def("layers", &Net::layers);
.def("blobs", &Net::blobs, py::return_value_policy::reference_internal)
.def("layers", &Net::layers, py::return_value_policy::reference_internal);
py::enum_<ncnn::BorderType>(m, "BorderType")
.value("BORDER_CONSTANT", ncnn::BorderType::BORDER_CONSTANT)