Ps: 目前本框架在持续开发中,由于是业余时间开发,更新速度不会很快。也欢迎有志之士一块维护本框架. 欢迎三连.
目前测试过的操作系统 (你可以自己测试下mac os)
windows | linux | mac | Loongnix | |
---|---|---|---|---|
checked | ||||
gpu |
目前测试过的CPU (其它U出现BUG可以在issue提出)
Intel i7 | raspberry 3B | raspberry 4B | Jeston NX | 龙芯 | |
---|---|---|---|---|---|
checked |
特点
已测试的网络
Yolo测试
Win10 MSVC 2017 I7-10700F (Linux平台大约比Win快10~20%,当然不是所有网络)
net | yolov3 | yolov3_tiny | yolov4 |
---|---|---|---|
time | 380ms | 50ms | 432ms |
ARM(Yolov3Tiny cpu)
cpu | raspberry 3B | raspberry 4B | Jeston NX |
---|---|---|---|
with neon asm | ? | 0.432s | ? |
Yolo GPU 测试
Ubuntu16.04 GCC Cuda10.1 GTX1080Ti
net | yolov3 | yolov3_tiny | yolov4 |
---|---|---|---|
time | 30ms | 8ms | 30ms |
Jetson NX
net | yolov3 | yolov3_tiny | yolov4 |
---|---|---|---|
time | 200ms | 20ms | 210ms |
Yolo GPU cuDnn FP16 测试
net | yolov3 | yolov4 |
---|---|---|
time | 115ms | 120ms |
Yolov5s GPU 测试
net | yolov5s | yolov5s_fp16 |
---|---|---|
time | 9.57ms | 8.57ms |
Mobilenet Yolo GPU cuDnn 测试
net | yoloface100k | yoloface500k | mobilenetv2_yolov3_nano | mobilenetv2_yolov3_lite |
---|---|---|---|---|
time | 7ms | 20ms | 20ms | 30ms |
DeepLabv3 GPU 测试
net | deeplabv3_resnet101 | deeplabv3_resnet50 |
---|---|---|
time | 22.51ms | 16.46ms |
依赖
视频教程(B站)
如何编译
ps. 你可以在CMakeLists.txt:52设置最大OMP占用线程, 默认是OMP使用所有CPU线程.(需取消勾选OMP_MAX_THREAD)
Windows平台编译(MSVC)
Linux(Ubuntu ) 注意: 构建NX的GPU版本取消勾选NEON OPENBLAS.
sudo apt-get install build-essential
sudo apt-get install qt5-default #可选
sudo apt-get install libqt5svg5-dev #可选
sudo apt-get install libopencv-dev #可选
sudo apt-get install libgl1-mesa-dev libglfw3-dev libglfw3 libglew-dev #可选
#config
sudo echo /usr/local/lib > /etc/ld.so.conf.d/usrlib.conf
sudo ldconfig
# build Msnhnet
git clone https://github.com/msnh2012/Msnhnet.git
mkdir build
cd Msnhnet/build
cmake -DCMAKE_BUILD_TYPE=Release ..
make -j4
sudo make install
vim ~/.bashrc # Last line add: export PATH=/usr/local/bin:$PATH
sudo ldconfig
测试
yolov3 D:/models
yolov3tiny_video D:/models
classify D:/models
使用MsnhnetViewer查看网络结构
各个封装的结构 ResBlock
Res2Block
AddBlock
ConcatBlock
如何转换模型
关于训练
Enjoy it! :D
感谢
Msnhnet参考了以下框架:
第三方库
加群交流
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。