当前位置： 移动技术网 > IT编程>脚本编程>Python > 目标检测yolo根据标签box坐标剪裁原图至新的文件夹

目标检测yolo根据标签box坐标剪裁原图至新的文件夹

2020年07月16日  | 移动技术网IT编程  | 我要评论
import osimport cv2import numpy as npfrom PIL import Imageimport matplotlib.pyplot as pltdef save(cropImg, framenum, tracker): pathnew = "../images1_cut/" #裁剪后文件夹 if (os.path.exists(pathnew)): cv2.imwrite(pathnew + framenum + '_' + 
import os
import cv2
import numpy as np
from PIL import Image
import matplotlib.pyplot as plt

def save(cropImg, framenum, tracker):  
    pathnew = "../images1_cut/"  #裁剪后文件夹
    if (os.path.exists(pathnew)):
        cv2.imwrite(pathnew + framenum + '_' + tracker + '.jpg', cropImg, [int(cv2.IMWRITE_JPEG_QUALITY), 100])
    else:
        os.makedirs(pathnew)
        cv2.imwrite(pathnew + framenum + tracker + '.jpg', cropImg, [int(cv2.IMWRITE_JPEG_QUALITY), 100])

f = open("../person.txt", "r")   #标签txt文件
lines = f.readlines()
for line in lines:
    li = line.split(' ')
    print(li[0], li[1], li[2], li[3], li[4], li[5])  #根据自己的txt内容格式进行修改
    filename = li[0] + '.jpg'
    img = cv2.imread("../images1/" + filename)  #原始图片文件夹

    a = int(float(li[3]))  # xmin 
    b = int(float(li[5]))  # xmax
    c = int(float(li[2]))  # ymin 
    d = int(float(li[4]))  # ymax 
    cropImg = img[a:b, c:d]  # 裁剪

    save(cropImg, li[0], str(li[1]).split('.')[1][:3])  #保留小数

本文地址：https://blog.csdn.net/mndlgzzd/article/details/107366878

您可能感兴趣的文章:

如您对本文有疑问或者有任何想说的，请 点击进行留言回复，万千网友为您解惑！

相关文章:

验证码：
最近更新的文章
大家感兴趣的文章
移动技术网