YOLOv5部署到web端詳細(xì)過程(flask+js簡單易懂)
更新時(shí)間：2023年04月26日 09:07:56 作者：帥帥帥.
YOLOv5是一個(gè)目標(biāo)檢測模型,Flask是一個(gè)Python的Web框架,下面這篇文章主要給大家介紹了關(guān)于YOLOv5部署到web端(flask+js簡單易懂)的相關(guān)資料,文中通過實(shí)例代碼介紹的非常詳細(xì),需要的朋友可以參考下
前言
最近，老板讓寫一個(gè)程序把yolov5檢測模型部署到web端，在網(wǎng)頁直接進(jìn)行目標(biāo)檢測。經(jīng)過1個(gè)星期的努力，終于實(shí)現(xiàn)基本功能??（累暈了）。網(wǎng)上查閱了很多資料，發(fā)現(xiàn)很多的技術(shù)比較復(fù)雜，當(dāng)然他們的頁面都比較漂亮，然后我打算自己寫一篇簡單記錄一下哈哈。進(jìn)入正題，該項(xiàng)目完全由Python的輕量級(jí)web框架flask完成，其實(shí)有一點(diǎn)js代碼，簡單易懂??
最終實(shí)現(xiàn)效果圖
先看項(xiàng)目的效果圖吧哈哈，比較辣雞，但是功能實(shí)現(xiàn)了！（后續(xù)會(huì)優(yōu)化頁面）
檢測結(jié)果顯示：
哈哈 可以看到檢測結(jié)果還是可以的??
后端實(shí)現(xiàn)
首先，寫了三個(gè)路由函數(shù)，分別是主界面、檢測函數(shù)、檢測結(jié)果顯示。如下所示
主界面
# 定義路由
@app.route('/', methods=['GET', 'POST'])
def upload():
if request.method == 'POST':
# 從表單中獲取上傳的文件
f = request.files['file']
global filename
filename = f.filename
global file_path
# 將文件保存到服務(wù)器本地
file_path = os.path.join(os.getcwd(), filename)
print(file_path)
f.save(file_path)
# 返回文件路徑
# return file_path
return render_template('index1.html')
檢測函數(shù)
#檢測函數(shù)
@app.route('/det', methods=['GET','POST'])
def my_flask_function():
#print('測試一下！')
opt = parse_opt()
main(opt)
# return jsonify({'message': 'Hello from Flask!'})
return render_template('123.html')檢測結(jié)果顯示
@app.route('/sh', methods=['GET', 'POST'])
def hello_world():
img_path = 'runs\\detect\\exp\\' + str(filename)
img_stream = return_img_stream(img_path)
return render_template('showimage.html', img_stream=img_stream)
#獲取圖片并展現(xiàn)到前端頁面
def return_img_stream(img_local_path):
"""
工具函數(shù):
獲取本地圖片流
:param img_local_path:文件單張圖片的本地絕對路徑
:return: 圖片流
"""
import base64
img_stream = ''
with open(img_local_path, 'rb') as img_f:
img_stream = img_f.read()
img_stream = base64.b64encode(img_stream).decode()
return img_stream
最后，我把detect文件中的函數(shù)，parse_out函數(shù)直接放到app.py文件中了，因?yàn)檫@樣傳遞圖片路徑比較方便。其中的main函數(shù)直接是調(diào)用的detect文件，如下所示：
# 檢測圖片的
def parse_opt():
parser = argparse.ArgumentParser()
parser.add_argument('--weights', nargs='+', type=str, default= 'yolov5s.pt', help='model path or triton URL')
#parser.add_argument('--source', type=str, default=0, help='file/dir/URL/glob/screen/0(webcam)')
parser.add_argument('--source', type=str, default= file_path, help='file/dir/URL/glob/screen/0(webcam)')
parser.add_argument('--data', type=str, default= 'models/yolov5s.yaml', help='(optional) dataset.yaml path')
parser.add_argument('--imgsz', '--img', '--img-size', nargs='+', type=int, default=[640], help='inference size h,w')
parser.add_argument('--conf-thres', type=float, default=0.25, help='confidence threshold')
parser.add_argument('--iou-thres', type=float, default=0.45, help='NMS IoU threshold')
parser.add_argument('--max-det', type=int, default=1000, help='maximum detections per image')
parser.add_argument('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')
parser.add_argument('--project', default= 'runs/detect', help='save results to project/name')
parser.add_argument('--name', default='exp', help='save results to project/name')
parser.add_argument('--exist-ok', action='store_true', help='existing project/name ok, do not increment')
parser.add_argument('--vid-stride', type=int, default=1, help='video frame-rate stride')
opt = parser.parse_args()
opt.imgsz *= 2 if len(opt.imgsz) == 1 else 1 # expand
#print_args(vars(opt))
args = parser.parse_args(args=[])
print(args)
return opt到這里，后端實(shí)現(xiàn)就基本寫完了。接下來，看看前端頁面 很簡單，我都不好意思寫??
前端實(shí)現(xiàn)
主要就兩個(gè)頁面，分別是主頁面index1.html和showimage.html。
主界面(index1.html）
index1.html就是主界面，顯示的一些操作按鈕。代碼如下：
<!DOCTYPE html>
<html lang="en">
<head>
<title>簡單實(shí)現(xiàn)</title>
<script src="https://cdn.bootcdn.net/ajax/libs/jquery/3.6.0/jquery.min.js"></script>
</head>
<body>
<h2>選擇圖片上傳檢測</h2>
<br />
<br />
<img id="imageDisplay" width="500" height="500" />
<br />
<form method="POST" enctype="multipart/form-data">
<input type="file" name="file">
<input type="submit" value="上傳">
</form>
<br />
<div id="imagePath"></div>
<script>
function displayImage() {
var fileInput = document.getElementById("imageFile");
var imagePath = document.getElementById("imagePath");
var imageDisplay = document.getElementById("imageDisplay");
var file = fileInput.files[0];
var reader = new FileReader();
reader.readAsDataURL(file);
reader.onload = function (e) {
imagePath.innerHTML = "Image Path: " + fileInput.value;
imageDisplay.src = e.target.result;
};
}
function Detect() {
$.ajax({
type: 'POST',
url: '/det',
success: function(response) {
console.log(response);
},
error: function(error) {
console.log(error);
}
});
}
function Show() {
var img = document.getElementById('img');
img.src = '/sh';
}
</script>
<button id="my-button" onclick="Detect()">檢測</button>
<a >
<button id="my-button1" onclick="Show()">顯示圖片</button>
</a>
</body>
</html>
顯示圖片界面（showimage.html）
然后就是 showiamge.html，該頁面是在點(diǎn)擊顯示圖片按鈕之后，進(jìn)行跳轉(zhuǎn)，然后顯示檢測結(jié)果的圖片。代碼如下所示：
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>檢測結(jié)果顯示</title>
</head>
<body>
<img style="width:500px " src="data:;base64,{{ img_stream }}">
</body>
</html>
總結(jié)
寫的比較潦草，代碼的功能不難理解，相對比較基礎(chǔ)。有什么問題可以在評論直接問。然后就是代碼基本上都放上來了，可以直接復(fù)制然后用，如果想直接要項(xiàng)目代碼，可以私信我（不收任何費(fèi)用）。
后續(xù)會(huì)優(yōu)化前端界面（等有時(shí)間了），因?yàn)檎娴奶罅恕?/strong>
到此這篇關(guān)于YOLOv5部署到web端的文章就介紹到這了,更多相關(guān)YOLOv5部署web端內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
在數(shù)字化辦公場景中,Word文檔作為承載結(jié)構(gòu)化信息的重要載體,常包含大量嵌入的超鏈接以關(guān)聯(lián)外部資源或?qū)崿F(xiàn)內(nèi)容交互,下面我們看看如何使用Python實(shí)現(xiàn)批量提取這些超鏈接吧2025-03-03
Python實(shí)現(xiàn)RLE格式與PNG格式互轉(zhuǎn)
在機(jī)器視覺領(lǐng)域的深度學(xué)習(xí)中，很多數(shù)據(jù)集的標(biāo)注文件使用RLE的格式。但是神經(jīng)網(wǎng)絡(luò)的輸入一定是一張圖片，為此必須把RLE格式的文件轉(zhuǎn)變?yōu)閳D像格式。本文將利用Python實(shí)現(xiàn)RLE格式與PNG格式互轉(zhuǎn)，感興趣的可以了解一下2022-08-08
pytorch加載預(yù)訓(xùn)練模型與自己模型不匹配的解決方案
這篇文章主要介紹了pytorch加載預(yù)訓(xùn)練模型與自己模型不匹配的解決方案，具有很好的參考價(jià)值，希望對大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2021-05-05
Python?的列表推導(dǎo)式，這個(gè)看似簡單的語法糖，實(shí)則內(nèi)含無限威力，本文將從基礎(chǔ)的概念認(rèn)識(shí)，到各類進(jìn)階的用法和操作，更深入地探討列表推導(dǎo)式，希望對大家有所幫助2023-06-06
python3實(shí)現(xiàn)點(diǎn)餐系統(tǒng)
這篇文章主要為大家詳細(xì)介紹了python3實(shí)現(xiàn)點(diǎn)餐系統(tǒng)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2019-01-01
Python實(shí)現(xiàn)四個(gè)經(jīng)典小游戲合集
這篇文章主要介紹了利用Python編寫一個(gè)經(jīng)典小游戲的合集，包括：貪吃蛇，掃雷，俄羅斯方塊，五子棋。感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下2021-12-12
解決Django no such table: django_session的問題
這篇文章主要介紹了解決Django no such table: django_session的問題，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-04-04
Python3以GitHub為例來實(shí)現(xiàn)模擬登錄和爬取的實(shí)例講解
在本篇內(nèi)容里小編給大家分享的是關(guān)于Python3以GitHub為例來實(shí)現(xiàn)模擬登錄和爬取的實(shí)例講解，需要的朋友們可以參考下。2020-07-07
這篇文章主要介紹了簡單的Python調(diào)度器Schedule詳解,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-08-08
這篇文章主要介紹了python的第三方庫pandas的相關(guān)知識(shí)，本文給大家介紹的非常詳細(xì)，具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2020-02-02