YoloAll V2發(fā)布,集成所有主流Yolo模型于一身

YoloAll項(xiàng)目簡(jiǎn)介
YoloAll是一個(gè)將當(dāng)前主流Yolo版本集成到同一個(gè)UI界面下的推理預(yù)測(cè)工具。可以迅速切換不同的yolo版本,并且可以針對(duì)圖片,視頻,攝像頭碼流進(jìn)行實(shí)時(shí)推理,可以很方便,直觀的對(duì)比不同版本的yolo的推理效果,耗時(shí)等。
自從YoloAll V1版本推出以后,得到了很多朋友的喜歡,大家也跟我一樣,被眾多的Yolo版本所困擾,有時(shí)為了比較兩個(gè)不同版本的Yolo的效果,往往需要花費(fèi)很長(zhǎng)時(shí)間搭建環(huán)境,并且很容易出錯(cuò),因此才有了這個(gè)開發(fā)這個(gè)YoloAll的想法,能夠非常方便的測(cè)試不同Yolo的效果!注意這個(gè)是測(cè)試推理效果的哦,不能進(jìn)行訓(xùn)練,如果想要非常方便的訓(xùn)練,推薦另外一個(gè)可視化的YoloX訓(xùn)練項(xiàng)目:YoloX EasyTrain:https://github.com/DL-Practise/YoloX_EasyTrain
YoloAll V2.0新特性
更全\更美\更強(qiáng)\更易用
增加了YoloV4,更新YoloV5到6.0版本,更新YoloX到0.1.1版本
重新設(shè)計(jì)了界面,美觀度提升,使用起來(lái)更加舒暢
增加了模型推理時(shí)候的參數(shù)配置
增加了模型自動(dòng)下載功能,方便下載與訓(xùn)練模型;
集成了使用手冊(cè),方便查閱;
將所有模型的依賴放到根目錄,安裝依賴更加方便
YoloAll安裝
操作系統(tǒng):Win10、Win7、Ubuntu16.04(其他操作系統(tǒng)沒(méi)有測(cè)試過(guò))
python環(huán)境:python3.7
cuda環(huán)境:cuda10.1(也可以不用cuda,使用cpu推理)
step1、下載YoloAll源碼
step2、解壓YoloAll,進(jìn)入到根目錄
cd? root/to/yoloall?
step3、升級(jí)pip
python -m pip install --upgrade pip
step4、安裝依賴項(xiàng)
pip install -r requirements.txt
YoloAll使用
啟動(dòng)YoloAll
python main_widget.py
界面簡(jiǎn)介
主要包括模型管理界面,配置界面,預(yù)測(cè)界面、消息界面和日志界面組成。
模型管理界面會(huì)將所有的模型以及子模型(例如YoloX包含s,m,l等子模型)
配置界面用于配置一些預(yù)測(cè)參數(shù),例如置信度閾值,nms閾值,圖片大小等
預(yù)測(cè)界面用于展現(xiàn)預(yù)測(cè)結(jié)果
消息界面展現(xiàn)YoloAll的升級(jí)信息,使用教程等
日志界面:展示一些關(guān)鍵的日志信息

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

下載預(yù)訓(xùn)練模型
由于預(yù)訓(xùn)練模型通常都很大,因此,下載的YoloAll中是不包含預(yù)訓(xùn)練模型的,需要通過(guò)鼠標(biāo)點(diǎn)擊模型管理界面中的子模型,例如點(diǎn)擊YoloX下面的yolox_l模型,會(huì)彈出提示框,提示預(yù)訓(xùn)練模型未下載,然后會(huì)在提示框中展示預(yù)訓(xùn)練模型的下載鏈接(有可能有多個(gè)下載鏈接),選擇一個(gè)網(wǎng)速不錯(cuò)的鏈接,從瀏覽器下載,下載完畢之后,根據(jù)提示框中的提示信息,放到對(duì)應(yīng)的文件夾下面。

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

此時(shí)可以點(diǎn)擊預(yù)測(cè)界面的Photo按鈕,選擇一張圖片進(jìn)行預(yù)測(cè),也可以點(diǎn)擊Video按鍵,選擇一段視頻文件進(jìn)行預(yù)測(cè),也可以點(diǎn)擊Camera按鈕,啟動(dòng)攝像頭進(jìn)行拍攝和預(yù)測(cè)。
修改預(yù)測(cè)參數(shù)
在預(yù)測(cè)過(guò)程中,如果需要進(jìn)行CPU、GPU的切換,或者修改預(yù)測(cè)參數(shù),例如修改圖像分辨率,置信度閾值,nms閾值等,可以在配置界面進(jìn)行修改的勾選或者修改,修改完后,點(diǎn)擊保存按鈕,軟件會(huì)重新創(chuàng)建模型,等模型創(chuàng)建成功后,可以繼續(xù)使用新的配置進(jìn)行預(yù)測(cè)。

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

YoloAll演示視頻(V2.0)
后續(xù)補(bǔ)上。。。