nnie yolov5初探
硬件hisi3519av100yolov5s移植,鉴于yolov5中的upsample,silu,maxpool等的op在nnie上不一致,做出如下修改。## 模型修改。### 参考官方的yolov5 4.0版本+ git clone -b v4.0 https://github.com/ultralytics/yolov5.git/### 修改处1, utils/activations.py 增加如下: class ReLU(nn.Module): @st...
原创
2021-09-17 13:21:29 ·
1395 阅读 ·
8 评论