目标检测数据增强：截取出bbox中的图片

最新推荐文章于 2024-06-25 07:28:53 发布

winnerziqi

最新推荐文章于 2024-06-25 07:28:53 发布

阅读量1.6k

点赞数 2

分类专栏： data augument 文章标签： python opencv 计算机视觉 java cv

本文链接：https://blog.csdn.net/wty98wzq/article/details/114364245

版权

crop_img文件的作用为截取出bbox中的图片

import os
import shutil
from os.path import join
import cv2
import glob

root_dir = "./fruit" #原始图片保存的位置
save_dir = "./bbox" #生成截取图片的位置

jpg_list = glob.glob(root_dir + "/*.jpg")


fo = open("dpj_small.txt", "w") #截取出来的图片位置

max_s = -1
min_s = 1000

for jpg_path in jpg_list: #遍历所有图片
    # jpg_path = jpg_list[3]
    txt_path = jpg_path.replace("jpg"