Recorte Python puede Oh! ! (5 líneas de código para resolver)

Aquí Insertar imagen Descripción

configuración de la instalación

python -m pip install paddlepaddle -i https://mirror.baidu.com/pypi/simple
  • Python en el entorno de prueba
Python 3.7.6 (tags/v3.7.6:43364a7ae0, Dec 19 2019, 00:42:30) [MSC v.1916 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import paddle.fluid
>>> paddle.fluid.install_check.run_check()
  • Si la pantalla de la consola de su paleta está instalado correctamente! Vamos a empezar con el aprendizaje profundo Paddle ahora en nombre de nuestra ya instalado con éxito. Además, también es necesario instalar paddlehub:
pip install -i https://mirror.baidu.com/pypi/simple paddlehub

recorte lograr

Recorte de la implementación del código es muy simple, más o menos dividido en los siguientes pasos:

  1. módulo de importación
  2. Cargar modelo
  3. Obtener una lista de archivos
  4. recorte

Los detalles de los puntos de código son los siguientes:

# 1、导入模块
import os
import paddlehub as hub

# 2、加载模型
humanseg = hub.Module(name='deeplabv3p_xception65_humanseg')

# 3、获取文件列表
# 图片文件的目录
path = 'D:/CodeField/Workplace/PythonWorkplace/PillowTest/11_yellow/img/'
# 获取目录下的文件
files = os.listdir(path)
# 用来装图片的
imgs = []
# 拼接图片路径
for i in files:
    imgs.append(path + i)
#抠图
results = humanseg.segmentation(data={'image':imgs})

Una vez finalizada la carrera, podemos ver el directorio humanseg_output el marco del proyecto, se tire una imagen buena almacenado en el directorio. Por supuesto, el código anterior obtenemos la operación de lista de archivos también se puede simplificar:

import os, paddlehub as hub
humanseg = hub.Module(name='deeplabv3p_xception65_humanseg')		# 加载模型
path = 'D:/CodeField/Workplace/PythonWorkplace/PillowTest/11_yellow/img/'	# 文件目录
files = [path + i for i in os.listdir(path)]	# 获取文件列表
results = humanseg.segmentation(data={'image':files})	# 抠图

Luego completamos las 5 líneas de esteras código de lote.

Publicado 84 artículos originales · ganado elogios 1 · vistas 2058

Supongo que te gusta

Origin blog.csdn.net/lxp_mocheng/article/details/104820091
Recomendado
Clasificación