切换版块
×
基础软件平台
PyTorch开发
TensorFlow开发
编解码及SDK开发
BANG语言与计算库
开发工具链
MagicMind开发
软件栈百科
云平台集成
硬件产品专区
MLU370系列AI加速卡
MLU270系列AI加速卡
MLU220系列AI加速产品
经验方案交流区
经验案例与实践分享
开发者服务
开发者活动
公告与版务
高校支持
《智能计算系统》
签到
版块
社区
文档
SDK下载
370系列
200系列
开发平台
官网首页
注册
登录
全部版块
基础软件平台
硬件产品专区
经验方案交流区
开发者服务
高校支持
发布新帖
登录/注册
LV.1
Jerry3062
36
积分
0
赞
3
帖子
13
回复
0
收藏
TA的动态
TA的帖子
TA的回复
ubuntu18 yolov5 编译的pytorch cambricon 运行量化模型失败
我的回复:#1HelloAI回复用你的代码,简单修改了部分代码后验证,并未发现结果错误的问题,修改的部分如下:1、models\common.py添加from utils.activations import SiLU然后替换33行的nn.SiLU()为SiLU():self.act = SiLU() if act is True else (act if isinstance(act, nn.Module) else nn.Identity())2、utils\torch_utils.py添加:from utils.activations import Hardswishfrom utils.activations import SiLU然后替换133行的nn.Hardswish为Hardswish:elif t in [Hardswish, nn.LeakyReLU, nn.ReLU, nn.ReLU6, SiLU]:3、detect.py的代码,去掉了所有half的配置,包括:61行,66、67行、112行、115行、128行4、然后去掉134行box_result = get_boxes(pred)的注释,然后print(box_result)的结果即为正确结果5、另外,量化的过程是在CPU运行了推理过程,所以推理结果依赖后处理nms,而MLU推理时,后处理已经替换为torch.ops.torch_mlu.yolov5_detection_output了,已经包含了后处理,不再需要nms了,这一定您的代码中已经体现出来了。您可再按照上述方法试下,谢谢。展开收到,感谢回复。暂时切换到了其他工作,完成后我会回来接着尝试。
0
Cambricon Pytorch编译安装问题?
我的回复:好的,谢谢。
0
pytorch yolov5在docker环境中推理结果不正确
我的回复:你好,代码是基于该仓库https://github.com/CambriconECO/Pytorch_Yolov5_Inference 我修改过的文件detect.py我将它的后缀名改为txt已经上传。detect.txt
0
Github
开发平台
文档中心
新手必读
官方微信
版权所有 © 2024 寒武纪 Cambricon.com 备案/许可证号:
京ICP备17003415号-1
关闭