來源：投稿 作者：ΔU

編輯：學(xué)姐

YoloAll項(xiàng)目簡(jiǎn)介

相信了解YOLO的小伙伴們一定都有這樣的困擾，目前YOLO各個(gè)版本數(shù)量非常多，不知道在實(shí)際場(chǎng)景中應(yīng)該選擇哪個(gè)YOLO版本。甚至有時(shí)為了比較兩個(gè)不同版本的YOLO的效果，需要自己花費(fèi)很長(zhǎng)的時(shí)間去搭建環(huán)境，下載權(quán)重訓(xùn)練模型，非常的浪費(fèi)精力，今天我就給大家分享一款GItHub上的開源項(xiàng)目----YOLOALL，YOLOALL能夠非常方便的測(cè)試不同版本的YOLO的檢測(cè)效果，并且可以針對(duì)圖片、視頻、攝像頭進(jìn)行預(yù)測(cè)，可以很方便直觀的對(duì)比不同版本YOLO的檢測(cè)效果和時(shí)間消耗。

下面來看一下該項(xiàng)目的具體教程吧！

YOLOALL安裝

系統(tǒng)要求：

操作系統(tǒng)：支持Win7、Win10、Ubuntu16+

Python環(huán)境：python3.7

Cuda環(huán)境：cuda10.1（也可以不用cuda，使用cpu預(yù)測(cè)）

安裝步驟：

1. 后臺(tái)回復(fù)YOLOALL獲取項(xiàng)目源碼

2.解壓YOLOALL，進(jìn)入到根目錄：cd /root/of/yoloall

3.升級(jí)pip：python -m pip install --upgrade pip

4. 安裝依賴項(xiàng)：pip install -r requirements.txt

YOLOALL使用

界面簡(jiǎn)介

預(yù)測(cè)界面：用于展現(xiàn)預(yù)測(cè)結(jié)果

日志界面：展示一些關(guān)鍵的日志信息

模型管理：用于選擇不同版本模型

配置及幫助：用于配置置信度，NMS閾值，圖片大小等參數(shù)

圖1：界面簡(jiǎn)介

啟動(dòng)YoloAll

第一次啟動(dòng)需要輸入命令：python main_widget.py

剛啟動(dòng)YOLOAll的時(shí)候，需要等待一段時(shí)間，此時(shí)在日志界面會(huì)有相關(guān)的提示信息，需要等到日志界面出現(xiàn)“加載模型結(jié)束”。此時(shí)模型管理界面會(huì)出現(xiàn)所有加載的模型，表示模型的加載已經(jīng)結(jié)束，可以進(jìn)行測(cè)試了。

預(yù)測(cè)

根據(jù)提示下載完預(yù)訓(xùn)練模型，并放到指定文件夾后，再次點(diǎn)擊模型管理界面的模型，就會(huì)開始創(chuàng)建模型，并顯示創(chuàng)建模型成功的界面。

點(diǎn)擊Photo按鈕：選擇一張圖片進(jìn)行預(yù)測(cè)，

點(diǎn)擊Video按鈕：選擇一段視頻文件進(jìn)行預(yù)測(cè)，

點(diǎn)擊Camera按鈕：啟動(dòng)攝像頭進(jìn)行拍攝和預(yù)測(cè)。

圖2：模型下載提示

修改參數(shù)

在預(yù)測(cè)過程中，如果需要進(jìn)行CPU、GPU的切換，或者修改預(yù)測(cè)參數(shù)，例如修改圖像分辨率，置信度閾值，NMS閾值等，可以在配置界面進(jìn)行修改的勾選或者修改，修改完后，點(diǎn)擊保存按鈕，軟件會(huì)重新創(chuàng)建模型，等模型創(chuàng)建成功后，可以繼續(xù)使用新的配置進(jìn)行預(yù)測(cè)。

圖3：修改參數(shù)頁面

預(yù)測(cè)結(jié)果

預(yù)測(cè)成功后，會(huì)在預(yù)測(cè)界面展示預(yù)測(cè)的信息，圖中的方框即為預(yù)測(cè)出來的目標(biāo)，紅色的文字是預(yù)測(cè)的速度和FPS。大家可以切換不同的模型，進(jìn)行預(yù)測(cè)結(jié)果以及耗時(shí)的對(duì)比。

圖4：YOLO v3檢測(cè)效果

圖5：YOLO v5檢測(cè)效果

項(xiàng)目地址：

https://github.com/DL-Practise/YoloAll/releases/tag/V2.0.1

演示視頻請(qǐng)關(guān)注【學(xué)姐帶你玩AI】公眾號(hào)觀看