tensorflow tfserving + django,搭建图像分类预测web服务

本文档介绍了如何结合Django web框架和TensorFlow Serving,搭建一个图像分类预测的服务。通过Django处理用户输入的图片URL,进行预处理,向tfserving发送请求,并显示预测结果。详细步骤包括Django容器的配置、项目创建、代码修改等,最终实现用户友好的Web接口。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

tensorflow tfserving + django,搭建图像分类预测web服务

前面介绍了使用docker和tfserving搭建一个预测服务,但是有时候需要对输入到网络的数据做一些预处理的工作,虽然可以在本地使用如Python代码处理完成后再请求预测,但是使用起来会不方便。有时候也可能有将预测的过程作为一个web服务,实现在网页上填写图片的地址信息,然后返回预测结果的需求。本篇主要基于之前搭建的tfserving预测服务,使用django (一个开源的python web框架,详细信息见官网 django) 搭建一个web服务,作为一个中转服务器,接受用户的输入,完成一些预处理工作,然后向tfserving请求预测结果,最后返回结果。全部源代码放在 github

另:使用docker+django+SSD搭建目标检测服务

先来看结果,下图为最终的页面:
在这里插入图片描述
点击这里访问,可以先试一试效果。输入图片的URL,点击predict,等待返回结果。

下面介绍主要的流程:

0. tfserving准备

这里使用的是TensorFlow官方给的一个关于tfserving的教程所带的模型,具体的Docker安装和tfserving容器搭建流程可以参考使用docker和tfserving搭建预测服务。该步骤为基础步骤,如果要完整实现该demo,需要首先完成此步骤。

1.获取django镜像

$docker pull django

2.启动一个django容器

将容器的8000端口映射到宿主机的8080端口(后面请求的时候用的就是8080端口),并进入终端

$docker run -p 8080:8000 -it django bash

在这里插入图片描述

3.创建一个django工程

$django-admin startproject app

在/usr/src/路径下执行该命令,会在当前目录下创建一个app文件夹及相关文件,此时的目录结构为&#

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值