YOLOv5部署，并实现Java客户端向Python服务端发送请求进行图片火苗检测

痛定思痛。 2024-03-17 22:43 35阅读 0赞

> 马上要参加计算机设计大赛，Web系统中开发了火苗实时检测功能。  
> 因为之前一直没有去做，所以今晚女娲补天式急速开发（三个小时），因此业务上难免有逻辑错误，代码也难免有问题。  
> 该文章仅记录一下该功能的核心代码，有错或者如何优化请大佬指出。

#### 模型检测效果展示： ####

![在这里插入图片描述][2e755059f17449d79c2d3e34c76936cd.jpeg_pic_center]  
![在这里插入图片描述][6283eb88bf3b49bb885c4110d5b4b5ed.jpeg_pic_center]

#### 业务流程图： ####

![在这里插入图片描述][af5e5e26d97841d69dbb9dda209edd76.png_pic_center]

#### Python服务端核心代码： ####

from flask import Flask, request, send_file
    from detect import parse_opt, main
    import shutil
    import urllib.request
    
    app = Flask(__name__)
    
    
    @app.route('/detect', methods=['POST'])
    def detect():
        image_url = request.data.decode('utf-8')
        output_path = 'data/images/fire/image.jpg'
        download_image(image_url, output_path)
    
        delete_folder('runs/detect')
        opt = parse_opt()
        main(opt)
    
        return send_file('runs/detect/exp/image.jpg', mimetype='image/jpg')
    
    
    def delete_folder(folder_path):
        try:
            shutil.rmtree(folder_path)
            print(f"文件夹 {
          folder_path} 删除成功")
        except Exception as e:
            print(f"删除文件夹 {
          folder_path} 失败: {
          str(e)}")
    
    
    def download_image(url, output_path):
        try:
            urllib.request.urlretrieve(url, output_path)
            print("图片保存成功！")
        except Exception as e:
            print(f"图片保存失败：{
          str(e)}")
    
    
    if __name__ == '__main__':
        app.run()

#### Java客户端核心代码： ####

package com.ruoyi.ai.fire;
    
    import com.ruoyi.sc.domain.QiniuFile;
    import com.ruoyi.sc.utils.QiniuUtils;
    import org.springframework.core.io.ByteArrayResource;
    import org.springframework.http.*;
    import org.springframework.web.client.RestTemplate;
    import org.springframework.web.multipart.MultipartFile;
    import org.springframework.mock.web.MockMultipartFile;
    
    /**
     * @Author: xinglibao
     * @Date: 2023/7/19 23:05
     */
    public class FireDetectService {
        
    
        public String detect(String url) {
        
            // 创建RestTemplate实例
            RestTemplate restTemplate = new RestTemplate();
    
            // 设置请求头
            HttpHeaders headers = new HttpHeaders();
            headers.setContentType(MediaType.TEXT_PLAIN);
    
            // 构建请求体
            HttpEntity<String> requestEntity = new HttpEntity<>(url, headers);
    
            // 发送请求
            ResponseEntity<byte[]> responseEntity = restTemplate.postForEntity("http://127.0.0.1:5000/detect", requestEntity, byte[].class);
    
            // 获取返回的图片文件
            byte[] imageBytes = responseEntity.getBody();
    
            // 将图片上传至七牛云
            ByteArrayResource resource = new ByteArrayResource(imageBytes);
            MultipartFile file = null;
            try {
        
                file = new MockMultipartFile("image.jpg", resource.getInputStream());
            } catch (Exception e) {
        
                e.printStackTrace();
            }
            QiniuFile qiniuFile = QiniuUtils.upload(file);
            return qiniuFile.getUrl();
    
            /*AjaxResult ajax = AjaxResult.success();
            ajax.put("url", qiniuFile.getUrl());
            ajax.put("fileName", qiniuFile.getFileName());
            ajax.put("newFileName", qiniuFile.getNewFileName());
            ajax.put("originalFilename", qiniuFile.getOriginalFilename());*/
        }
    
        public static void main(String[] args) {
        
            System.out.println(new FireDetectService().detect("http://lygshengben.com/typora/nnnfsdfds.jpg"));
        }
    
    }

#### 系统前端界面展示： ####

![在这里插入图片描述][bf0d5e2dc1644fe58f536dcd0bbb260d.jpeg_pic_center]  
![在这里插入图片描述][bddb2fec50594b57b108284dc84d36f6.jpeg_pic_center]

#### 附：SpringBoot整合七牛云实现文件上传——附Utils工具类（若依整合七牛云）： ####

[SpringBoot整合七牛云实现文件上传——附Utils工具类（若依整合七牛云）][SpringBoot_Utils]

[2e755059f17449d79c2d3e34c76936cd.jpeg_pic_center]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/14/caedf93fe4644abdace8f50124ca447c.png
[6283eb88bf3b49bb885c4110d5b4b5ed.jpeg_pic_center]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/14/4ccbe7ff74e744efa31a20443584ff24.png
[af5e5e26d97841d69dbb9dda209edd76.png_pic_center]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/14/9bf63d648b4f427f95f2897584b6b0ba.png
[bf0d5e2dc1644fe58f536dcd0bbb260d.jpeg_pic_center]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/14/4eb57ccf408749db84fc74f344eb90ed.png
[bddb2fec50594b57b108284dc84d36f6.jpeg_pic_center]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/14/afe70a80f10c42c3885dc2993a4ffda5.png
[SpringBoot_Utils]: https://blog.csdn.net/weixin_45953673/article/details/129645320