# unet_mxnet2ncnn
Train mxnet unet and run it on ncnn.
How to use:
1. Get dataset and put it in the some directory. Adjust files under mxnet-unet/data based on dataset location.
2. cd mxnet-unet; python3 trainunet.py
3. Get inference model by running: python3 train2infer.py
4. Use ncnn tool 'mxnet-ncnn' to get ncnn model
Need to patch tools/mxnet/mxnet2ncnn.cpp for ncnn. Set num_group to 1 in if for "Deconvolution" op. But I think it will fix also if designate it in mxnet network.
5. Run ncnn-unet.
Install OpenCV first.
cd ncnn-unet
Edit CMakefiles to fit your enviroment.
mkdir build
cmake ..; make
./unet your_jpg_file
没有合适的资源?快使用搜索试试~ 我知道了~
资源推荐
资源详情
资源评论
收起资源包目录
算法部署_使用NCNN部署MXnet-Unet图像分割算法_优质算法部署项目实战.zip (12个子文件)
算法部署_使用NCNN部署MXnet-Unet图像分割算法_优质算法部署项目实战
ncnn-unet
CMakeLists.txt 558B
src
main.cpp 5KB
models
convert.sh 222B
mxnet-unet
trainunet.py 4KB
unetdataiter.py 5KB
data
seg_train.txt 3.25MB
seg_test.txt 429KB
predict.py 2KB
unet.py 7KB
check.py 3KB
train2infer.py 1KB
README.md 664B
共 12 条
- 1
资源评论
__AtYou__
- 粉丝: 3506
- 资源: 2175
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
最新资源
资源上传下载、课程学习等过程中有任何疑问或建议,欢迎提出宝贵意见哦~我们会及时处理!
点击此处反馈
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功