环境安装
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