主頁 >  其他 > 使用Fastai開發和部署影像分類器應用

使用Fastai開發和部署影像分類器應用

2020-11-08 18:50:24 其他

作者|[email protected]
編譯|Flin
來源|analyticsvidhya

介紹

Fastai是一個流行的開源庫,用于學習和練習機器學習以及深度學習,杰里米·霍華德(Jeremy Howard)和蕾切爾·托馬斯(Rachel Thomas)創立了 fast.ai, 其目標是使深度學習資源更容易獲得,fast.ai中提供的所有詳盡的資源,例如課程,軟體和研究論文都是完全免費的,

2020年8月,fastai_v2發布,該版本有望更快,更靈活地實作深度學習框架,在2020 fastai課程結合了機器學習和深入學習的核心概念,它還向用戶介紹了模型生產和部署的重要方面,

在本文中,我將討論fast.ai初學者課程的前三課中介紹的有關建立快速簡單的影像分類模型的技術,在構建模型的同時,你還將學習如何輕松地為模型開發Web應用程式并將其部署到生產環境,

本文將遵循Jeremy在其課程中所采用的自上而下的教學方法,你將首先學習有關訓練影像分類器的知識,稍后,將解釋有關用于分類的模型的細節,要理解本文,你必須具備Python知識,因為fastai是用Python撰寫并基于PyTorch構建的,建議你在Google Colab或Gradient中運行此代碼,因為我們需要GPU訪問權限,而且fastai可以輕松安裝在這兩個平臺上,

安裝,匯入和加載資料集

!pip install -Uqq fastbook
import fastbook
fastbook.setup_book()

from fastbook import *
from fastai.vision.widgets import *

安裝fastai并匯入必要的庫,如果你使用的是Colab,則必須提供對Google云端硬碟的訪問權限以保存檔案和影像,你可以從Kaggle和Bing影像搜索等來源下載任何影像資料集,Fast.ai也有大量的影像集合,我在本文中使用了來自 https://github.com/ieee8023/covid-chestxray-dataset 的一組胸部X射線影像,

path = Path ('/content/gdrive/My Drive/Covid19images')

將資料集位置的路徑保存在Path()物件中,如果使用fast.ai資料集,則可以使用以下代碼:

path = untar_data(URLs.PETS)/'images'

這將從fastai PETS資料集集合中下載并提取影像,

檢查影像路徑并顯示資料集中的一些樣本影像,我已經為此使用了Python Imaging Library(PIL),

path.ls
from PIL import Image
img = Image.open(path'/train/covid/1-s2.0-S1684118220300682-main.pdf-002-a2.png')
print(img.shape)
img.to_thumb(128,128)

在此影像分類問題中,我將訓練模型,以將X射線影像分類為COVID或No COVID類,預處理資料集已放置在單獨的COVID和No COVID檔案夾中(來源:ChristianTutivénGálvez),

如果你使用的是fast.ai資料集,請使用以下函式根據寵物的名稱對影像進行分組:

def is_cat(x): return x[0].isupper()

PETS是貓和狗影像的集合,Cat圖片標有大寫的第一個字母,因此很容易對其進行分類,

影像變換

影像變換是訓練影像模型的關鍵步驟,它也稱為資料擴充,為了避免模型過擬合,必須進行影像變換,可以使用多種方法來轉換影像,例如調整大小,裁剪,壓縮和填充,但是,壓縮和填充會搶奪影像中的原始資訊,并分別添加其他像素,因此,隨機調整影像大小會產生良好的效果,

如以下示例所示,在此方法中,在每個時期都對每個影像的隨機區域進行采樣,這使模型可以了解每個影像的更多細節,從而獲得更高的準確性,

要記住的另一個要點是,始終僅變換訓練影像,而不修改驗證影像,在fastai庫中,默認情況下會處理此問題,

item_tfms=Resize(128, ResizeMethod.Squish))
item_tfms=Resize(128, ResizeMethod.Pad, pad_mode='zeros')
item_tfms=RandomResizedCrop(128, min_scale=0.3) - 30% of the image area is zoomed by specifying 0.3

Fastai庫通過aug_transforms函式提供了一組標準的擴充,如果影像尺寸均勻,可以批量應用,節省大量訓練時間,

tfms = aug_transforms(do_flip = True, flip_vert = False, mult=2.0)

fastai中的DataLoaders類非常便于存盤用于訓練和驗證模型的各種物件,如果要自定義訓練期間要使用的物件,則可以將DataBlock類與DataLoaders結合使用,

data= https://www.cnblogs.com/panchuangai/archive/2020/11/08/ImageDataLoaders.from_folder(path,train ="train", valid_pct=0.2, item_tfms=Resize(128), batch_tfms=tfms, bs = 30, num_workers = 4)

如果你在圖元檔案中定義了影像標簽,則可以使用DataBlock將影像和標簽分為兩個不同的塊,如下面的代碼片段所示,將定義的資料塊與資料加載器功能一起使用以訪問影像,

Data = https://www.cnblogs.com/panchuangai/archive/2020/11/08/DataBlock( blocks=(ImageBlock, CategoryBlock), get_items=get_image_files, 
splitter=RandomSplitter(valid_pct=0.2, seed=42), get_y=parent_label, item_tfms=Resize(128))
dls = Data.dataloaders(path)

模型訓練

為了訓練該影像資料集,使用了預訓練的CNN模型,這種方法稱為轉移學習,杰里米(Jeremy)建議使用預先訓練的模型,以加快訓練速度并提高準確性,這尤其適用于計算機視覺問題,

learn = cnn_learner(data, resnet34, metrics=error_rate)
learn.fine_tune(4)
 

使用ResNet34體系結構,并根據錯誤率驗證結果,由于使用預先訓練的模型進行訓練,因此使用微調方法而不是對模型進行擬合,

你可以運行更多時期,并查看模型的性能,選擇正確的時期數以避免過擬合,

你可以嘗試使用準確性(準確性= 1-錯誤率)來驗證模型性能,而不是使用error_rate,兩者都用于驗證模型的輸出,在此示例中,保留了20%的資料用于驗證,因此,該模型將僅對80%的資料進行訓練,這是檢查任何機器學習模型性能的非常關鍵的一步,你也可以通過更改ResNet層(選項為18、50、101和152)來運行此模型,除非你有一個大型資料集將產生準確的結果,否則這可能再次導致過擬合,

驗證模型性能

模型性能可以通過不同的方式進行驗證,一種流行的方法是使用混淆矩陣,矩陣的對角線值指示每種類別的正確預測,而其他單元格值指示許多錯誤的預測,

interp = ClassificationInterpretation.from_learner(learn)
interp.plot_confusion_matrix()

Fastai提供了一個有用的功能,可以根據最高丟失率查看錯誤的預測,該函式的輸出指示每個影像的預測標簽,目標標簽,丟失率和概率值,高概率表示模型具有較高的置信度,它在0到1之間變化,高丟失率表示模型性能有多差,

interp.plot_top_losses(5, nrows=1, figsize = (25,5))

另一個很棒的Fastai功能, ImageClassifierCleaner(GUI),它可以通過洗掉故障影像或重命名其標簽來清除故障影像,這非常有助于資料預處理,從而提高了模型的準確性,

杰里米(Jeremy)建議在對影像進行基本訓練后再運行此功能,因為這可以了解資料集中例外的種類,

from fastai.vision.widgets import *
cleaner = ImageClassifierCleaner(learn)
cleaner

保存和部署模型

訓練完模型并對結果滿意后,就可以部署模型了,要將模型部署到生產環境中,你需要保存模型體系結構以及對其進行訓練的引數,為此,使用了匯出方法,匯出的模型另存為PKL檔案,該檔案是pickle(Python模塊)創建的檔案,

learn.export()

從匯出的檔案中創建一個推理學習器,該學習器可用于將模型部署為應用程式,推理學習器一次預測一個新影像的輸出,預測回傳三個引數:預測類別,預測類別的索引以及每個類別的概率,

learn_inf = load_learner(path/'export.pkl')
learn_inf.predict("img")

(‘noCovid’, tensor(1), tensor([5.4443e-05, 9.9995e-01])) – prediction

有多種方法可以創建用于部署模型的Web應用程式,最簡單的方法之一是使用作為GUI組件的IPython小部件在Jupyter notebook中為應用程式創建所需的物件,

from fastai.vision.widgets import *
btn_upload = widgets.FileUpload()
out_pl = widgets.Output()
lbl_pred = widgets.Label()

設計應用程式元素后,請使用像Web應用程式一樣運行Jupyter notebook的Voila來部署模型,它洗掉所有單元格輸入,僅顯示模型輸出,要將notebook作為VoilàWeb應用程式查看,請將瀏覽器URL中的“notebook”一詞替換為“ voila/render”,必須在包含受過訓練的模型和IPython小部件的同一notebook中安裝和執行Voila,

!pip install voila
!jupyter serverextension enable voila --sys-prefix

結論

就這樣,你已經使用fastai庫構建并部署了一個很酷的影像分類器應用程式,只需八個步驟!這還只是我在本文中展示的冰山一角,有更多的fastai組件可用于與NLP和計算機視覺相關的各種深度學習用例,你可以探索這些組件,

以下是fastai學習資源,以及我的git repo,其中包含本文中解釋的影像分類器的代碼和影像,

  • Covid19 X射線影像分類器:包含本文中討論的完整代碼和資料集

    • https://github.com/RajiRai/FastAI/blob/master/Covid19_prediction_using_X_ray.ipynb
  • 涵蓋了fast.ai課程中教授的所有課程

    • https://github.com/fastai/fastbook
  • 涵蓋完整的fastai API檔案

    • https://docs.fast.ai/
  • Fast.ai社區論壇

    • https://forums.fast.ai/

原文鏈接:https://www.analyticsvidhya.com/blog/2020/10/develop-and-deploy-an-image-classifier-app-using-fastai/

歡迎關注磐創AI博客站:
http://panchuang.net/

sklearn機器學習中文官方檔案:
http://sklearn123.com/

歡迎關注磐創博客資源匯總站:
http://docs.panchuang.net/

轉載請註明出處,本文鏈接:https://www.uj5u.com/qita/206951.html

標籤:其他

上一篇:使用基于GAN的過采樣技術提高非平衡COVID-19死亡率預測的模型準確性

下一篇:win10 安裝pytorch gpu 及 解決報錯“OSError: [WinError 126] 找不到指定的模塊

標籤雲
其他(157675) Python(38076) JavaScript(25376) Java(17977) C(15215) 區塊鏈(8255) C#(7972) AI(7469) 爪哇(7425) MySQL(7132) html(6777) 基礎類(6313) sql(6102) 熊猫(6058) PHP(5869) 数组(5741) R(5409) Linux(5327) 反应(5209) 腳本語言(PerlPython)(5129) 非技術區(4971) Android(4554) 数据框(4311) css(4259) 节点.js(4032) C語言(3288) json(3245) 列表(3129) 扑(3119) C++語言(3117) 安卓(2998) 打字稿(2995) VBA(2789) Java相關(2746) 疑難問題(2699) 细绳(2522) 單片機工控(2479) iOS(2429) ASP.NET(2402) MongoDB(2323) 麻木的(2285) 正则表达式(2254) 字典(2211) 循环(2198) 迅速(2185) 擅长(2169) 镖(2155) 功能(1967) .NET技术(1958) Web開發(1951) python-3.x(1918) HtmlCss(1915) 弹簧靴(1913) C++(1909) xml(1889) PostgreSQL(1872) .NETCore(1853) 谷歌表格(1846) Unity3D(1843) for循环(1842)

熱門瀏覽
  • 網閘典型架構簡述

    網閘架構一般分為兩種:三主機的三系統架構網閘和雙主機的2+1架構網閘。 三主機架構分別為內端機、外端機和仲裁機。三機無論從軟體和硬體上均各自獨立。首先從硬體上來看,三機都用各自獨立的主板、記憶體及存盤設備。從軟體上來看,三機有各自獨立的作業系統。這樣能達到完全的三機獨立。對于“2+1”系統,“2”分為 ......

    uj5u.com 2020-09-10 02:00:44 more
  • 如何從xshell上傳檔案到centos linux虛擬機里

    如何從xshell上傳檔案到centos linux虛擬機里及:虛擬機CentOs下執行 yum -y install lrzsz命令,出現錯誤:鏡像無法找到軟體包 前言 一、安裝lrzsz步驟 二、上傳檔案 三、遇到的問題及解決方案 總結 前言 提示:其實很簡單,往虛擬機上安裝一個上傳檔案的工具 ......

    uj5u.com 2020-09-10 02:00:47 more
  • 一、SQLMAP入門

    一、SQLMAP入門 1、判斷是否存在注入 sqlmap.py -u 網址/id=1 id=1不可缺少。當注入點后面的引數大于兩個時。需要加雙引號, sqlmap.py -u "網址/id=1&uid=1" 2、判斷文本中的請求是否存在注入 從文本中加載http請求,SQLMAP可以從一個文本檔案中 ......

    uj5u.com 2020-09-10 02:00:50 more
  • Metasploit 簡單使用教程

    metasploit 簡單使用教程 浩先生, 2020-08-28 16:18:25 分類專欄: kail 網路安全 linux 文章標簽: linux資訊安全 編輯 著作權 metasploit 使用教程 前言 一、Metasploit是什么? 二、準備作業 三、具體步驟 前言 Msfconsole ......

    uj5u.com 2020-09-10 02:00:53 more
  • 游戲逆向之驅動層與用戶層通訊

    驅動層代碼: #pragma once #include <ntifs.h> #define add_code CTL_CODE(FILE_DEVICE_UNKNOWN,0x800,METHOD_BUFFERED,FILE_ANY_ACCESS) /* 更多游戲逆向視頻www.yxfzedu.com ......

    uj5u.com 2020-09-10 02:00:56 more
  • 北斗電力時鐘(北斗授時服務器)讓網路資料更精準

    北斗電力時鐘(北斗授時服務器)讓網路資料更精準 北斗電力時鐘(北斗授時服務器)讓網路資料更精準 京準電子科技官微——ahjzsz 近幾年,資訊技術的得了快速發展,互聯網在逐漸普及,其在人們生活和生產中都得到了廣泛應用,并且取得了不錯的應用效果。計算機網路資訊在電力系統中的應用,一方面使電力系統的運行 ......

    uj5u.com 2020-09-10 02:01:03 more
  • 【CTF】CTFHub 技能樹 彩蛋 writeup

    ?碎碎念 CTFHub:https://www.ctfhub.com/ 筆者入門CTF時時剛開始刷的是bugku的舊平臺,后來才有了CTFHub。 感覺不論是網頁UI設計,還是題目質量,賽事跟蹤,工具軟體都做得很不錯。 而且因為獨到的金幣制度的確讓人有一種想去刷題賺金幣的感覺。 個人還是非常喜歡這個 ......

    uj5u.com 2020-09-10 02:04:05 more
  • 02windows基礎操作

    我學到了一下幾點 Windows系統目錄結構與滲透的作用 常見Windows的服務詳解 Windows埠詳解 常用的Windows注冊表詳解 hacker DOS命令詳解(net user / type /md /rd/ dir /cd /net use copy、批處理 等) 利用dos命令制作 ......

    uj5u.com 2020-09-10 02:04:18 more
  • 03.Linux基礎操作

    我學到了以下幾點 01Linux系統介紹02系統安裝,密碼啊破解03Linux常用命令04LAMP 01LINUX windows: win03 8 12 16 19 配置不繁瑣 Linux:redhat,centos(紅帽社區版),Ubuntu server,suse unix:金融機構,證券,銀 ......

    uj5u.com 2020-09-10 02:04:30 more
  • 05HTML

    01HTML介紹 02頭部標簽講解03基礎標簽講解04表單標簽講解 HTML前段語言 js1.了解代碼2.根據代碼 懂得挖掘漏洞 (POST注入/XSS漏洞上傳)3.黑帽seo 白帽seo 客戶網站被黑帽植入劫持代碼如何處理4.熟悉html表單 <html><head><title>TDK標題,描述 ......

    uj5u.com 2020-09-10 02:04:36 more
最新发布
  • 2023年最新微信小程式抓包教程

    01 開門見山 隔一個月發一篇文章,不過分。 首先回顧一下《微信系結手機號資料庫被脫庫事件》,我也是第一時間得知了這個訊息,然后跟蹤了整件事情的經過。下面是這起事件的相關截圖以及近日流出的一萬條資料樣本: 個人認為這件事也沒什么,還不如關注一下之前45億快遞資料查詢渠道疑似在近日復活的訊息。 訊息是 ......

    uj5u.com 2023-04-20 08:48:24 more
  • web3 產品介紹:metamask 錢包 使用最多的瀏覽器插件錢包

    Metamask錢包是一種基于區塊鏈技術的數字貨幣錢包,它允許用戶在安全、便捷的環境下管理自己的加密資產。Metamask錢包是以太坊生態系統中最流行的錢包之一,它具有易于使用、安全性高和功能強大等優點。 本文將詳細介紹Metamask錢包的功能和使用方法。 一、 Metamask錢包的功能 數字資 ......

    uj5u.com 2023-04-20 08:47:46 more
  • vulnhub_Earth

    前言 靶機地址->>>vulnhub_Earth 攻擊機ip:192.168.20.121 靶機ip:192.168.20.122 參考文章 https://www.cnblogs.com/Jing-X/archive/2022/04/03/16097695.html https://www.cnb ......

    uj5u.com 2023-04-20 07:46:20 more
  • 從4k到42k,軟體測驗工程師的漲薪史,給我看哭了

    清明節一過,盲猜大家已經無心上班,在數著日子準備過五一,但一想到銀行卡里的余額……瞬間心情就不美麗了。最近,2023年高校畢業生就業調查顯示,本科畢業月平均起薪為5825元。調查一出,便有很多同學表示自己又被平均了。看著這一資料,不免讓人想到前不久中國青年報的一項調查:近六成大學生認為畢業10年內會 ......

    uj5u.com 2023-04-20 07:44:00 more
  • 最新版本 Stable Diffusion 開源 AI 繪畫工具之中文自動提詞篇

    🎈 標簽生成器 由于輸入正向提示詞 prompt 和反向提示詞 negative prompt 都是使用英文,所以對學習母語的我們非常不友好 使用網址:https://tinygeeker.github.io/p/ai-prompt-generator 這個網址是為了讓大家在使用 AI 繪畫的時候 ......

    uj5u.com 2023-04-20 07:43:36 more
  • 漫談前端自動化測驗演進之路及測驗工具分析

    隨著前端技術的不斷發展和應用程式的日益復雜,前端自動化測驗也在不斷演進。隨著 Web 應用程式變得越來越復雜,自動化測驗的需求也越來越高。如今,自動化測驗已經成為 Web 應用程式開發程序中不可或缺的一部分,它們可以幫助開發人員更快地發現和修復錯誤,提高應用程式的性能和可靠性。 ......

    uj5u.com 2023-04-20 07:43:16 more
  • CANN開發實踐:4個DVPP記憶體問題的典型案例解讀

    摘要:由于DVPP媒體資料處理功能對存放輸入、輸出資料的記憶體有更高的要求(例如,記憶體首地址128位元組對齊),因此需呼叫專用的記憶體申請介面,那么本期就分享幾個關于DVPP記憶體問題的典型案例,并給出原因分析及解決方法。 本文分享自華為云社區《FAQ_DVPP記憶體問題案例》,作者:昇騰CANN。 DVPP ......

    uj5u.com 2023-04-20 07:43:03 more
  • msf學習

    msf學習 以kali自帶的msf為例 一、msf核心模塊與功能 msf模塊都放在/usr/share/metasploit-framework/modules目錄下 1、auxiliary 輔助模塊,輔助滲透(埠掃描、登錄密碼爆破、漏洞驗證等) 2、encoders 編碼器模塊,主要包含各種編碼 ......

    uj5u.com 2023-04-20 07:42:59 more
  • Halcon軟體安裝與界面簡介

    1. 下載Halcon17版本到到本地 2. 雙擊安裝包后 3. 步驟如下 1.2 Halcon軟體安裝 界面分為四大塊 1. Halcon的五個助手 1) 影像采集助手:與相機連接,設定相機引數,采集影像 2) 標定助手:九點標定或是其它的標定,生成標定檔案及內參外參,可以將像素單位轉換為長度單位 ......

    uj5u.com 2023-04-20 07:42:17 more
  • 在MacOS下使用Unity3D開發游戲

    第一次發博客,先發一下我的游戲開發環境吧。 去年2月份買了一臺MacBookPro2021 M1pro(以下簡稱mbp),這一年來一直在用mbp開發游戲。我大致分享一下我的開發工具以及使用體驗。 1、Unity 官網鏈接: https://unity.cn/releases 我一般使用的Apple ......

    uj5u.com 2023-04-20 07:40:19 more