yolo v8 + flask部署到云服务器，以及问题记录

智商不够_熬夜来凑

已于 2024-08-27 21:44:52 修改

阅读量1k

点赞数 3

CC 4.0 BY-SA版权

文章标签： YOLO pytorch python flask

于 2024-08-08 10:49:32 首次发布

本文链接：https://blog.youkuaiyun.com/gzyh_tech/article/details/140988903

环境安装

1、运行项目报错：no python application found, check your startup logs for errors

在云服务器pytorch版本安装错了，安装了GPU版本，需要安装CPU版本

# CPU only 使用下面这段代码避免出现第二个错误
pip install torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 --index-url https://download.pytorch.org/whl/cpu

2、运行项目报错：RuntimeError: operator torchvision::nms does not exist

检查发现pytorch中torchvision版本不匹配：

卸载重装对应匹配版本：

# CPU only
pip install torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 --index-url https://download.pytorch.org/whl/cpu

3、后端python文件编写，涉及到读写文件、模型预测、以及获取结果分析，转换yolo预测结果为指定的json格式数据。

import os
import numpy as np
import torch.hub
#导入Yolov8，需要提前安装ultralytics库
from ultralytics import YOLO
from flask import Flask
#运用Python的flask类实现与前台信息的交互
from flask import request
from flask import send_file
import base64
import cv2
import time
import json
from pathlib import Path

app = Flask(__name__)
# 调用训练好的模型
model = YOLO('./best.pt')
@app.route('/', methods=['GET', 'POST'])
def uploads():
    # 拿到变量img对应的图片
    img = request.files.get('img')
    if img:
        # 重命名
        name = 'img.jpg'
        # 保存
        img.save(os.path.join('./img', name))
        fileName = "./img/"+name
        # results = model.pre