沈阳外贸网站建设,寻找外贸客户的网站,盱眙县建设局网站,保定满城网站建设NanoDet与YOLO一样#xff0c;都是目标检测算法中的经典之作#xff0c;在NanoDet-plus发布之后#xff0c;很多CVer都争先恐后地进行尝试#xff0c;感受其与NanoDet的不同。以上的截图已经将NanoDet-plus的优点完全概括#xff0c;这里就不再赘述。新的plus版本精度和速… NanoDet与YOLO一样都是目标检测算法中的经典之作在NanoDet-plus发布之后很多CVer都争先恐后地进行尝试感受其与NanoDet的不同。以上的截图已经将NanoDet-plus的优点完全概括这里就不再赘述。新的plus版本精度和速度都有提高与其他轻量级模型如YOLOv5-n、YOLOX-Nano等相比1ms的延时换来了30%精度的提升较为简单的训练辅助模块和易于部署的模型也得到了广大CVer的好评不多说我们开始尝试一下吧。下图是NanoDet-plus与其他目标检测算法的对比来源是Github的NanoDet仓库。 一、软硬件配置
1、软件Windows 10专业版、Visual Studio 2019安装CUDA时需要、CUDA11.1、Pytorch1.8.1、Pycharm专业版、Python3.9.7。CUDA安装教程连接目标检测第3步如何在Windows 10系统下安装CUDA更新时间2022.03.22_liO_Oil的博客-CSDN博客_cuda windows
2、硬件主要是显卡要求N卡2060。CPU、内存等没有具体要求。
二、下载源代码 Nanodet-plus源码下载地址GitHub - RangiLyu/nanodet at v1.0.0-alpha-1 这里使用的是最新的V1.0.0-alpha-1版本。
三、解压并创建虚拟环境
1、解压源码压缩包并使用Pycharm创建对应的虚拟环境。 上图中创建的虚拟环境已有基础的依赖项接下来我们使用pip命令安装其他依赖。
2、使用“pip install -r requirements.txt”安装其他依赖项。 打开Pycharm终端路径前显示venv表明我们已进入先前创建好的虚拟环境如果你的显示和图片中显示的不一致那么请看这篇博文如何设置Pycharm在打开Terminal终端时自动进入虚拟环境_liO_Oil的博客-CSDN博客_pycharm terminal 进入虚拟环境 接着输入pip命令
pip install -r requirements.txt
如果觉得较慢可以指定软件源为清华源进行下载命令为
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple以上的命令默认安装最新的PytorchCPU版所以我们需要进行更换换为能使用显卡且与CUDA版本相契合的PytorchGPU版。Pytorch的下载地址为PyTorch 这里我使用的是1.8.1所以需要到先前的版本里找下载命令为 pip install torch1.8.1cu111 torchvision0.9.1cu111 torchaudio0.8.1 -f https://download.pytorch.org/whl/torch_stable.html 安装完Pytorch1.8.1的小伙伴可能会在结尾看到红色的报错说是pytorch-lighting要求更高版本的Pytorch这时你可以将pytorch-lighting的版本降为1.6.0的就可以了。
3、最后一步
使用命令行运行setup.py即可完成nanodet-plus的搭建。
python setup.py develop 四、运行naodet-plus检测图片
推理图片自然需要权重文件下载地址还是GitHub - RangiLyu/nanodet at v1.0.0-alpha-1 我下载的是红框圈中的权重文件你们可试试其他的。其中的weight是推理时用到的权重文件训练时会用到checkpoint文件可以一起下载下来。下载好权重文件之后在nanodet-1.0.0-alpha-1的目录下创建一个名为“weights”文件夹来存储权重文件。接下来进行图片的推理我将YOLOv5的bus.jpg作为测试图片复制到nanodet-1.0.0-alpha-1目录下如下图所示。
在Pycharm里打开终端输入如下命令进行推理
python demo/demo.py image --config ./config/nanodet-plus-m-1.5x_416.yml --model weights/nanodet-plus-m-1.5x_416.pth --path bus.jpg
推理图片、视频或开启摄像头推理的命令行大家自行修改和尝试。 注意.yml是NanoDet-plus的配置文件.pth是权重文件这两个文件要一一对应哦下面就是检测出的结果 五、结语 最近Windows 11在逐步覆盖我个人认为此款操作系统可能不如Windows 10稳定与众多软件的兼容性较差所以还是安装稳定的Windows 10好除此以外像Pytorch、CUDA、Python等也是装较为稳定的好没必要更新到最新版。如果在搭建环境过程中遇到难以解决的问题欢迎大家在评论区留言或私信我。
