![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
华为atlas
文章平均质量分 59
watersink
你怎么对这个世界,世界就会还你一个怎么样的它
展开
-
基于华为atlas的unet分割模型探索
使用工具Netron查看模型结构,确定模型输入节点名称为input.1,输出节点名称为/outc/conv/Conv。华为atlas的参考案例细节不到位,步骤缺失较多,摸索困难,代码写法较差,信创化道路任重而道远。模型输入为572*572*3,输出为572*572*2。分割目标分别为,0:背景,1:汽车。Unet模型使用官方基于kaggle。原创 2024-03-05 16:58:17 · 668 阅读 · 2 评论 -
基于华为atlas的分类模型实战
修改mobilenetv3.py中网络结构,模型选用MobileNetV3_Small模型,网络输出节点增加softmax层,将原始的return self.linear4(out)修改为return F.softmax(self.linear4(out), dim=-1)分类模型选用基于imagenet训练的MobileNetV3模型,分类类别为1000类。模型的输出node增加softmax成功。模型的输入node名称为input.1。实现转化onnx代码,原创 2024-02-27 16:29:02 · 835 阅读 · 4 评论 -
华为atlas300安装教程
【代码】华为atlas300安装教程。原创 2023-12-20 11:52:01 · 783 阅读 · 0 评论 -
基于华为atlas的烟火检测实战
实现信创化的docker file用于生成docker image,初始系统选择openeuler-20.09系统,docker file文件内容如下,将yolov5的best.onnx模型拷贝到当前目录,进行onnx转化为om,输出yolov5_add_bs1_fp16.om。基于flask实现烟火检测算法的http服务,然后实现视频解码-AI识别-结果绘制于视频上进行视频编码的业务代码。最终效果如下,上边为业务代码、左下角为流媒体引擎代码、右下角为AI服务代码、中间为AI实时视频识别效果。原创 2023-12-20 11:17:33 · 994 阅读 · 0 评论