百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 热门文章 > 正文

AI技术「车辆分割」开发使用体验

bigegpt 2024-08-09 11:09 2 浏览

一、需求描述

从实拍图中自动分割出汽车图像,实现自动批量抠图,可进一步对背景图像进行替换、合成、虚化处理,用于新车、二手车宣传图制作。

二、使用攻略

说明:本文采用C# 语言,开发环境为.Net Core 2.1,采用在线API接口方式实现。

(1)平台接入

目前处于邀测阶段,不能直接在控制台调用,可通过QQ群(659268104)联系群管、或提交工单申请开通测试权限。

(2)接口文档

文档地址:https://ai.baidu.com/docs#/ImageClassify-API/3e953ab4

接口描述:传入单帧图像,检测图像中的车辆,以小汽车为主,识别车辆的轮廓范围,与背景进行分离,返回分割后的二值图、灰度图、前景抠图,支持多个车辆、车门打开、后备箱打开、机盖打开、正面、侧面、背面等各种拍摄场景。

请求说明

返回说明

(3)源码共享

3.1-根据 API Key 和 Secret Key 获取 AccessToken

 /// 
 /// 获取百度access_token
 /// 
 /// API Key
 /// Secret Key
 /// 
 public static string GetAccessToken(string clientId, string clientSecret)
 {
 string authHost = "https://aip.baidubce.com/oauth/2.0/token";
 HttpClient client = new HttpClient();
 List> paraList = new List>();
 paraList.Add(new KeyValuePair("grant_type", "client_credentials"));
 paraList.Add(new KeyValuePair("client_id", clientId));
 paraList.Add(new KeyValuePair("client_secret", clientSecret));
 HttpResponseMessage response = client.PostAsync(authHost, new FormUrlEncodedContent(paraList)).Result;
 string result = response.Content.ReadAsStringAsync().Result;
 JObject jo = (JObject)JsonConvert.DeserializeObject(result);
 string token = jo["access_token"].ToString();
 return token;
 }

3.2-调用API接口获取识别结果

1、在Startup.cs 文件 的 Configure(IApplicationBuilder app, IHostingEnvironment env) 方法中开启虚拟目录映射功能

 string webRootPath = HostingEnvironment.WebRootPath;//wwwroot目录
 app.UseStaticFiles(new StaticFileOptions
 {
 FileProvider = new PhysicalFileProvider(
 Path.Combine(webRootPath, "Uploads", "BaiduAIs")),
 RequestPath = "/BaiduAIs"
 });

2、 建立Index.cshtml文件

2.1 前台代码: 由于html代码无法原生显示,只能简单说明一下:

主要是一个form表单,需要设置属性enctype="multipart/form-data",否则无法上传图片;

form表单里面有四个控件:

一个Input:type="file",asp-for="FileUpload" ,上传图片用;

一个Input:type="submit",asp-page-handler="VehicleSeg" ,提交并返回识别结果。

一个img:src="@Model.curPath",显示需要识别的图片。

一个img:src="@Model.imgProcessPath",显示车辆分割后的前景抠图。

最后显示后台 msg 字符串列表信息,如果需要输出原始Html代码,则需要使用@Html.Raw()函数。

2.2 后台代码:

 [BindProperty]
 public IFormFile FileUpload { get; set; }
 [BindProperty]
 public string ImageUrl { get; set; }
 private readonly IHostingEnvironment HostingEnvironment;
 public List msg = new List();
 public string curPath { get; set; }
 public string imgProcessPath { get; set; }
 public ImageProcessModel(IHostingEnvironment hostingEnvironment)
 {
 HostingEnvironment = hostingEnvironment;
 }
 public async Task OnPostVehicleSegAsync()
 {
 
 if (FileUpload is null)
 {
 ModelState.AddModelError(string.Empty, "本地图片!");
 }
 if (!ModelState.IsValid)
 {
 return Page();
 }
 msg = new List();
 string webRootPath = HostingEnvironment.WebRootPath;//wwwroot目录
 string fileDir = Path.Combine(webRootPath, "Uploads//BaiduAIs//");
 string imgName = await UploadFile(FileUpload, fileDir);
 string fileName = Path.Combine(fileDir, imgName);
 string imgBase64 = GetFileBase64(fileName);
 curPath = Path.Combine("/BaiduAIs/", imgName);//需在Startup.cs 文件 的 Configure(IApplicationBuilder app, IHostingEnvironment env)方法中开启虚拟目录映射功能
 string result = GetImageJson(imgBase64, “你的API KEY”, “你的SECRET KEY”, "foreground");
 JObject jo = (JObject)JsonConvert.DeserializeObject(result);
 try
 {
 string imageProcessBase64 = jo["foreground"].ToString();
 msg.Add("车辆分割");
 msg.Add("log_id:" + jo["log_id"].ToString());
 string processImgName = Guid.NewGuid().ToString("N")+ ".png";
 string imgSavedPath = Path.Combine(webRootPath, "Uploads//BaiduAIs//", processImgName);
 imgProcessPath = Path.Combine("BaiduAIs//", processImgName);
 await GetFileFromBase64(imageProcessBase64, imgSavedPath);
 }
 catch (Exception e)
 {
 msg.Add(result);
 }
 return Page();
 }
 /// 
 /// 上传文件,返回文件名
 /// 
 /// 文件上传控件
 /// 文件绝对路径
 /// 
 public static async Task UploadFile(IFormFile formFile, string fileDir)
 {
 if (!Directory.Exists(fileDir))
 {
 Directory.CreateDirectory(fileDir);
 }
 string extension = Path.GetExtension(formFile.FileName);
 string imgName = Guid.NewGuid().ToString("N") + extension;
 var filePath = Path.Combine(fileDir, imgName);
 using (var fileStream = new FileStream(filePath, FileMode.Create, FileAccess.Write))
 {
 await formFile.CopyToAsync(fileStream);
 }
 return imgName;
 }
 /// 
 /// 返回图片的base64编码
 /// 
 /// 文件绝对路径名称
 /// 
 public static String GetFileBase64(string fileName)
 {
 FileStream filestream = new FileStream(fileName, FileMode.Open);
 byte[] arr = new byte[filestream.Length];
 filestream.Read(arr, 0, (int)filestream.Length);
 string baser64 = Convert.ToBase64String(arr);
 filestream.Close();
 return baser64;
 }
 /// 
 /// 文件base64解码
 /// 
 /// 文件base64编码
 /// 生成文件路径
 public static async Task GetFileFromBase64(string base64Str, string outPath)
 {
 var contents = Convert.FromBase64String(base64Str);
 using (var fs = new FileStream(outPath, FileMode.Create, FileAccess.Write))
 {
 fs.Write(contents, 0, contents.Length);
 fs.Flush();
 }
 }
 /// 
 /// 图像识别Json字符串
 /// 
 /// 图片base64编码
 /// API Key
 /// Secret Key
 /// 
 public static string GetImageJson(string strbaser64, string clientId, string clientSecret, string type)
 {
 string token = GetAccessToken(clientId, clientSecret);
 string host = "https://aip.baidubce.com/rest/2.0/image-classify/v1/vehicle_seg?access_token=" + token;
 Encoding encoding = Encoding.Default;
 HttpWebRequest request = (HttpWebRequest)WebRequest.Create(host);
 request.Method = "post";
 request.KeepAlive = true;
 string str = "image=" + HttpUtility.UrlEncode(strbaser64)+"&type=" + type;
 byte[] buffer = encoding.GetBytes(str);
 request.ContentLength = buffer.Length;
 request.GetRequestStream().Write(buffer, 0, buffer.Length);
 HttpWebResponse response = (HttpWebResponse)request.GetResponse();
 StreamReader reader = new StreamReader(response.GetResponseStream(), Encoding.Default);
 string result = reader.ReadToEnd();
 return result;
 }

三、效果测试

1、页面:

2、识别结果:

2.1

2.2

2.3

2.4

四、测试结果

总体来说,车辆分割效果的好坏,跟照片中车辆环境还是有较大的关系的,如果车辆跟周围环境区别比较明细,则分割效果相当好,如果车辆跟周围环境区别不太明显,则分割效果就不稳定了。可以尝试优化当车辆跟周围环境较为相似时,如何采取更好的方法进行车辆分割:比如,可以先识别出当前车辆类型,然后根据该车辆类型的模型图,去进行分割等。

另外,可以加入一下功能,增强车辆分割技术:

1、自动补全功能

首先,关于前景抠图的轮廓处理,目前好像对其进行了一定的模糊处理,但是这种处理方法不是很好,如果能够使用AI的自动学习能力,根据输入的图片,搜索相似的车辆图片,然后对其进行自动补全操作,比如一辆车,它的一个轮胎被东西遮挡住了,如果单纯分割,分割后的图片中的车子是少一半轮胎的,这时候,如果能够结合AI学习技术,根据相似的车辆图片,对那缺失的一半轮胎进行修图补全,形成一个完整的车辆图片,那么,最后的抠图就有更大的价值了。

2、效果美化

对车辆图片进行补全后,若能够根据相似图,对抠图进行一定的美化功能(类似现在流行的美颜相机),那么美化后的抠图就是一张相当好的效果图,这样的话,就可以直接拿来当宣传图使用了。当然,美化训练,可以采取对大量的车辆图片进行打分,最后获取高分车辆图片,然后让AI学习如何将图像往高分图方向美化,这样美化后的图片就比较符合大众要求了。

3、多图录入,形成三维效果图

另外,如果能够实现对一辆车,进行前、后、左、右多方位拍图并输入系统,最后根据输入的图片,结合该车辆已存在的设计图等进行模拟计算,输出该车的三维效果图的话,运用前景就很大了。

3.1、结合百度AR技术,显示三维效果图

比如,可以结合百度AR技术,将车辆三维效果图显示到自己想要显示的地方,这样的话,不仅可以让汽车销售员直接出去推销车辆,在汽车销售介绍的时候,显示三维效果图让顾客能够全方向的了解车辆,还能让汽车爱好者收集自己喜欢的车辆,并跟车友们很好的进行分享交流。

3.2、结合3D打印技术,打印三维车辆模型

还可以结合3D打印技术,直接打印出车辆的模型。如果能够对一辆车进行多方位拍照,然后运用百度AI技术进行3D建模,使用3D打印技术直接打印出该车辆的模型,那是一件很了不起的事情。3D模型不仅可以当作小孩子们的玩具,也可以方便汽车爱好者们研究,还有利于汽车销售,让顾客有更加直观的了解。

原本链接:https://ai.baidu.com/forum/topic/show/955827

相关推荐

【Docker 新手入门指南】第十章:Dockerfile

Dockerfile是Docker镜像构建的核心配置文件,通过预定义的指令集实现镜像的自动化构建。以下从核心概念、指令详解、最佳实践三方面展开说明,帮助你系统掌握Dockerfile的使用逻...

Windows下最简单的ESP8266_ROTS_ESP-IDF环境搭建与腾讯云SDK编译

前言其实也没啥可说的,只是我感觉ESP-IDF对新手来说很不友好,很容易踩坑,尤其是对业余DIY爱好者搭建环境非常困难,即使有官方文档,或者网上的其他文档,但是还是很容易踩坑,多研究,记住两点就行了,...

python虚拟环境迁移(python虚拟环境conda)

主机A的虚拟环境向主机B迁移。前提条件:主机A和主机B已经安装了virtualenv1.主机A操作如下虚拟环境目录:venv进入虚拟环境:sourcevenv/bin/active(1)记录虚拟环...

Python爬虫进阶教程(二):线程、协程

简介线程线程也叫轻量级进程,它是一个基本的CPU执行单元,也是程序执行过程中的最小单元,由线程ID、程序计数器、寄存器集合和堆栈共同组成。线程的引入减小了程序并发执行时的开销,提高了操作系统的并发性能...

基于网络安全的Docker逃逸(docker)

如何判断当前机器是否为Docker容器环境Metasploit中的checkcontainer模块、(判断是否为虚拟机,checkvm模块)搭配学习教程1.检查根目录下是否存在.dockerenv文...

Python编程语言被纳入浙江高考,小学生都开始学了

今年9月份开始的新学期,浙江省三到九年级信息技术课将同步替换新教材。其中,新初二将新增Python编程课程内容。新高一信息技术编程语言由VB替换为Python,大数据、人工智能、程序设计与算法按照教材...

CentOS 7下安装Python 3.10的完整过程

1.安装相应的编译工具yum-ygroupinstall"Developmenttools"yum-yinstallzlib-develbzip2-develope...

如何在Ubuntu 20.04上部署Odoo 14

Odoo是世界上最受欢迎的多合一商务软件。它提供了一系列业务应用程序,包括CRM,网站,电子商务,计费,会计,制造,仓库,项目管理,库存等等,所有这些都无缝集成在一起。Odoo可以通过几种不同的方式进...

Ubuntu 系统安装 PyTorch 全流程指南

当前环境:Ubuntu22.04,显卡为GeForceRTX3080Ti1、下载显卡驱动驱动网站:https://www.nvidia.com/en-us/drivers/根据自己的显卡型号和...

spark+python环境搭建(python 环境搭建)

最近项目需要用到spark大数据相关技术,周末有空spark环境搭起来...目标spark,python运行环境部署在linux服务器个人通过vscode开发通过远程python解释器执行代码准备...

centos7.9安装最新python-3.11.1(centos安装python环境)

centos7.9安装最新python-3.11.1centos7.9默认安装的是python-2.7.5版本,安全扫描时会有很多漏洞,比如:Python命令注入漏洞(CVE-2015-2010...

Linux系统下,五大步骤安装Python

一、下载Python包网上教程大多是通过官方地址进行下载Python的,但由于国内网络环境问题,会导致下载很慢,所以这里建议通过国内镜像进行下载例如:淘宝镜像http://npm.taobao.or...

centos7上安装python3(centos7安装python3.7.2一键脚本)

centos7上默认安装的是python2,要使用python3则需要自行下载源码编译安装。1.安装依赖yum-ygroupinstall"Developmenttools"...

利用本地数据通过微调方式训练 本地DeepSeek-R1 蒸馏模型

网络上相应的教程基本都基于LLaMA-Factory进行,本文章主要顺着相应的教程一步步实现大模型的微调和训练。训练环境:可自行定义,mac、linux或者window之类的均可以,本文以ma...

【法器篇】天啦噜,库崩了没备份(天啦噜是什么意思?)

背景数据库没有做备份,一天突然由于断电或其他原因导致无法启动了,且设置了innodb_force_recovery=6都无法启动,里面的数据怎么才能恢复出来?本例采用解析建表语句+表空间传输的方式进行...