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

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

原创

最新推荐文章于 2024-06-25 07:28:53 发布 · 1.7k 阅读

19 ·

CC 4.0 BY-SA版权

文章标签：

#python #opencv #计算机视觉 #java #cv

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",