當(dāng)前位置：首頁(yè) > 人文社科 > 生活经验 >内容正文

生活经验

ubuntu下载安装MaskRCNN-benchmark

發(fā)布時(shí)間：2023/11/27 生活经验 36 豆豆

生活随笔收集整理的這篇文章主要介紹了 ubuntu下载安装MaskRCNN-benchmark 小編覺(jué)得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.

在window下，配個(gè)環(huán)境，一堆錯(cuò)誤，一周多都沒(méi)解決。換到ubuntu下，不到一天就配好了= =
強(qiáng)烈建議直接去ubuntu下配置，千萬(wàn)別在window環(huán)境下配置。

一、下載anaconda，創(chuàng)建虛擬環(huán)境

下載anaconda的步驟讀者可以去百度，很簡(jiǎn)單。隨后創(chuàng)建虛擬環(huán)境

conda create -n pytorch python=3.6.0
&emps;

二、下載CUDA與CUDNN

首先要更新驅(qū)動(dòng)，隨后進(jìn)行安裝。安裝步驟可參考博主之前的博客。

https://blog.csdn.net/weixin_43828675/article/details/115708033?spm=1001.2014.3001.5501
https://blog.csdn.net/weixin_43828675/article/details/115710652?spm=1001.2014.3001.5501
?

三、安裝maskrcnn-benchmark

筆者的環(huán)境信息

CUDA：10.0
CUDNN:7.6.5
pytorch：1.2.0

去pytorch官網(wǎng)下載相應(yīng)的安裝包

https://pytorch.org/get-started/previous-versions/

安裝cocoapi
安裝coco數(shù)據(jù)集有關(guān)的api
進(jìn)入自己的目錄下，進(jìn)入虛擬環(huán)境

source activate pytorch
git clone https://github.com/cocodataset/cocoapi.git
cd cocoapi/PythonAPI
python setup.py build_ext install

安裝apex
這一步是最容易錯(cuò)的，因?yàn)楣倬W(wǎng)上下的版本很可能與自己的電腦的GCC不匹配，需要進(jìn)行回退
apex是由nvidia維護(hù)的一個(gè)支持混合精度分布式訓(xùn)練的第三方pytorch擴(kuò)展

這里不加sudo的話，很可能無(wú)法下載

sudo git clone https://github.com/NVIDIA/apex.git

降低apex的版本

cd apex
sudo git checkout f3a960f80244cf9e80558ab30f7f7e8cbf03c0a0

改變apex文件夾的權(quán)限

sudo chmod -R 777 ../apex

安裝

python setup.py install --cuda_ext --cpp_ext

安裝Mask-RCNN

cd ..
git clone https://github.com/facebookresearch/maskrcnn-benchmark.git
cd maskrcnn-benchmark
python setup.py build develop

運(yùn)行demo

進(jìn)入到demo文件夾下

運(yùn)行

python3 webcam.py --min-image-size 300 MODEL.DEVICE cuda

運(yùn)行效果如下所示：

即使有GPU，運(yùn)行速度也很慢，一幀基本要0.6s

總結(jié)

以上是生活随笔為你收集整理的ubuntu下载安装MaskRCNN-benchmark的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。