python 编写http服务，接受图片并返回识别结果

业务流程就是客户端post待检测的数据，如

datas = {'file': img_str, 'classes': ['wcgz', 'wcaqm' ], 'vis_thresh':0.3}

其中‘file’为图片数据， ‘classes’为需要识别的对象名称，‘vis_thresh’为置信度。

服务器接受到后，检测发来的图片，如果识别出了待识别的对象则返回它的区域以及其他有用信息。

下面主要介绍数据收发这一块。其中图片发送借鉴于https://blog.csdn.net/zhong1213/article/details/103982782?depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-4&utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-4

注意图片的编解码

客户端：

# -*- coding: utf-8 -*-
import requests
import json
import numpy as np
import cv2
import base64

def getByte(path):
    with open(path, 'rb') as f:
        img_byte = base64.b64encode(f.read())
    img_str = img_byte.decode('ascii') 
    return img_str 


img_str = getByte('./1.jpg')


datas = {'file': img_str, 'classes': ['wcgz', 'wcaqm' ], 'vis_thresh':0.3}


req = json.dumps(datas)
r = requests.post("http://127.0.0.1:8000", data = req)


print(r.text)

服务端：

# -*- coding: utf-8 -*-
import codecs
from http.server import HTTPServer, BaseHTTPRequestHandler
from socketserver import ThreadingMixIn
#import DeviceFalutModels
import time
import json
import sys
sys.stdout = codecs.getwriter("utf-8")(sys.stdout.detach())


import cv2
import numpy as np
import base64

#服务端地址
host = ('localhost', 8000)

#ftp地址
ftpPath = 'ftp://192.168.0.136:21/result/'

class Resquest(BaseHTTPRequestHandler):
    def handler(self):
        print("data:", self.rfile.readline().decode())
        self.wfile.write(self.rfile.readline())
 
    def do_GET(self):
        print(self.requestline)
        self.send_response(200)
        self.send_header('Content-type', 'application/json')
        self.end_headers()
        self.wfile.write(json.dumps(data).encode())
        
    #接受post请求
    def do_POST(self):
        #print(self.headers)
        #print(self.command)
        #读取数据
        req_datas = self.rfile.read(int(self.headers['content-length']))
        #print(req_datas.decode())
    
        req = json.loads(req_datas.decode())
        #print(req)
        #检测
        result = Detection(req)
        
        self.send_response(200)
        self.send_header('Content-type', 'application/json')
        self.end_headers()
        
        #返回结果
        self.wfile.write(json.dumps(result).encode('utf-8'))
        

class ThreadingHttpServer(ThreadingMixIn, HTTPServer):
    pass

#检测
def Detection(msg):
    
    img_decode_as = msg['file'].encode('ascii') 
    img_decode = base64.b64decode(img_decode_as) 
    img_np_ = np.frombuffer(img_decode, np.uint8)
    img = cv2.imdecode(img_np_, cv2.COLOR_RGB2BGR) 


    #解析参数
    labelList = msg['classes']
    threshold = msg['vis_thresh']
    print(labelList, threshold)
    
    if threshold < 0.1:
        threshold = 0.1
        
    imgName = get_current_time() + '.jpg' 
    imgPath = './recvPics/' + imgName
    
    #判断图片是否有效
    try:
        cv2.imwrite(imgPath, img)
    except:
        print('img broken!')
        return {"code": "503", "result":'null', 'msg': '未处理的异常' }
    else:
        print('recvpic path: ', imgPath)
      
    #common_res = DeviceFalutModels.run_infence(imgsname= [imgPath], threshold, contestantId="1")
    #for test
    common_res = [['wcgz', 0.61, [984, 38, 1454, 551]], ['wcaqm', 0.51, [347, 12, 619, 326]], ['wcaqm', 0.57, [1148, 39, 1349, 288]]]
    #common_res = []
    print(common_res)
    
    if len(common_res) == 0:
        print("recognized nothing")
    
    result = []
    id = 0
    #在所有结果中查找指定标签
    for i in range(len(common_res)):
        for j in range(len(labelList)):
            res = {}
            if common_res[i][0] == labelList[j]:
                id = id + 1
                res['id'] = id
                res['label'] = labelList[j]
                res['type'] = parseFaultType(labelList[j])
                res['scores'] = float(common_res[i][1])
                if labelList[j] == 'kgg_ybh' or labelList[j] == 'kgg_ybf':
                    res['isAlarm'] = 0
                else:
                    res['isAlarm'] = 1
                    
                res['bbox'] = common_res[i][2]
                result.append(res)    
            
    
    #print(result)
    
    #保存在本地的图片文件名称
    strImgFile = "./resultPics/" + imgName
    print(strImgFile)
    
    RetData = {}
    if len(result) == 0:
        RetData["code"] = 210
        RetData["result"] = 'null'
        RetData["msg"] = '未识别到缺陷'
        
    else:
        RetData["code"] = 200
        RetData["result"] = result
        #ftp路径
        RetData["img_url"] = ftpPath + imgName
        RetData["img_path"] = strImgFile
        RetData["msg"] = '识别成功'
        
    print('RetData: ', RetData)
    return RetData
    
if __name__ == '__main__':
    myServer = ThreadingHttpServer(host, Resquest)

    print("Starting http server, listen at: %s:%s" % host)
    myServer.serve_forever()

服务器这里涉及到了一个多线程HTTPServer，参考https://blog.csdn.net/legend_x/article/details/14127675

这里接受到的图片数据保存到本地的时候，发现大小变了，这其实是cv::imwrite()的一个图像质量参数的问题

可参考：https://www.cnblogs.com/skyfsm/p/7136709.html