Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
YuxiangJohn 364ab940d2 | 3 years ago | |
---|---|---|
CMake | 3 years ago | |
Common | 3 years ago | |
DstEngine | 3 years ago | |
InferEngine | 3 years ago | |
include | 3 years ago | |
py | 3 years ago | |
pybind11 @ f980d76d38 | 3 years ago | |
src | 3 years ago | |
test | 3 years ago | |
.gitattributes | 3 years ago | |
.gitignore | 3 years ago | |
.gitmodules | 3 years ago | |
CMakeLists.txt | 3 years ago | |
LICENSE | 3 years ago | |
README.md | 3 years ago | |
build.sh | 3 years ago |
a simple, efficient, easy-to-use HUAWEI Ascend310 Atlas wrapper for cnn with c++ and python api. You will be able use atlas wrapper to fast demo or deploy your FP16 model with few lines of code!
//c++
// create Net
atlas::Net *net = new atlas::Net(modelPath, enginePath, deviceId);
// get Net information
net->getNetInfo()
// inferece
// accept input images as vector<cv::Mat> or vector<uint8_t> with vector<int> for image shape.
// recieve the output and outputDims from atlas::NetResult
net->inference(opencv_img_vec, results);
// net->inference(uint8_t_img_vec, image_shape_vec, results);
// destroy net
delete net;
# python
# import
import pyatlas
# create net
net = pyatlas.net(modelPath, enginePath, deviceId)
# get net information
netInfo = net.getNetInfo()
# inference
result = net.DoInference(img, imgSize)
Atlas DDK B893+ and driver
opencv_world 3.4+
for python api, python 2.x/3.x and numpy in needed
Make sure you had install dependencies list above.
# clone project and submodule
git clone --recurse-submodules ...
cd atlaswrapper
bash build.sh A300 #(options: A500, A200)
then you can intergrate it into your own project with libatlasWrapper.so and Net.h, for python module, you get pyatlas.so
Please follow the details from the documents of Atlas300
Please check the c++ & python demo on test folder.
AtlasWrapper is released under the [Apache 2.0 license].
该项目开源了一种针对华为海思Ascend 310芯片Atlas 的包装器。该算法简单、高效、易于使用,提供Python的应用接口。 使用该算法,只需几行代码就可以快速演示Atlas 或者部署FP16 模型。
C++ CMake C Text Shell other
Dear OpenI User
Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.
For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》