GitHub开源的ImageAI 库:几行代码可实现目标对象识别
目錄???????
1、圖像預(yù)測
2、物體檢測
3、視頻對象檢測和跟蹤
ImageAI是一個Python庫,旨在使開發(fā)人員能夠使用簡單的幾行代碼構(gòu)建具有自包含深度學(xué)習(xí)和計(jì)算機(jī)視覺功能的應(yīng)用程序和系統(tǒng)。?
ImageAI以簡潔為基礎(chǔ),?支持一系列最先進(jìn)的機(jī)器學(xué)習(xí)算法,用于圖像預(yù)測,自定義圖像預(yù)測,物體檢測,視頻檢測,視頻對象跟蹤和圖像預(yù)測培訓(xùn)。ImageAI目前支持使用在ImageNet-1000數(shù)據(jù)集上訓(xùn)練的4種不同機(jī)器學(xué)習(xí)算法進(jìn)行圖像預(yù)測和訓(xùn)練。ImageAI還支持使用在COCO數(shù)據(jù)集上訓(xùn)練的RetinaNet,YOLOv3和TinyYOLOv3進(jìn)行對象檢測,視頻檢測和對象跟蹤。
最終,ImageAI將為計(jì)算機(jī)視覺的更廣泛和更專業(yè)化方面提供支持,包括但不限于特殊環(huán)境和特殊領(lǐng)域的圖像識別。
ImageAI可實(shí)現(xiàn)的功能如下:
1、圖像預(yù)測
ImageAI提供4種不同的算法和模型類型來執(zhí)行圖像預(yù)測,并在ImageNet-1000數(shù)據(jù)集上進(jìn)行訓(xùn)練。提供用于圖像預(yù)測的4種算法包括SqueezeNet,ResNet,InceptionV3和DenseNet。
GitHub網(wǎng)站:https://github.com/OlafenwaMoses/ImageAI/tree/master/imageai/Prediction
2、物體檢測
ImageAI提供了非常方便和強(qiáng)大的方法來對圖像執(zhí)行對象檢測并從圖像中提取每個對象。對象檢測類為RetinaNet,YOLOv3和TinyYOLOv3提供支持,可選擇調(diào)整最先進(jìn)的性能或?qū)崟r處理。
GitHub網(wǎng)站:https://github.com/OlafenwaMoses/ImageAI/blob/master/imageai/Detection
3、視頻對象檢測和跟蹤
ImageAI提供了非常方便和強(qiáng)大的方法來在視頻中執(zhí)行對象檢測并跟蹤特定對象。提供的視頻對象檢測類僅支持當(dāng)前最先進(jìn)的RetinaNet,但具有調(diào)整最佳性能或?qū)崟r處理的選項(xiàng)。
另外用戶也可以自定義模型,訓(xùn)練屬于自己的識別模型,廢話少說,將ImageAI 庫GitHub網(wǎng)站奉上:https://github.com/OlafenwaMoses/ImageAI#object-detection?
總結(jié)
以上是生活随笔為你收集整理的GitHub开源的ImageAI 库:几行代码可实现目标对象识别的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 4 行 Python 代码获取所在城市天
- 下一篇: GitHub上共享的简单易用 Tenso