本文主要是介绍从数据集中挑选红色区域大于某一阈值的图片,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
算法训练时需要对图片进行过滤,改脚本用来从数据集中挑选中图片中红色区域大于某一阈值的图片。
#!usr/bin/env python
#-*- coding:utf-8 _*-import os
import shutilimport numpy as np
import cv2lower_red = np.array([0, 127, 128]) # 红色阈值下界
higher_red = np.array([10, 255, 255]) # 红色阈值上界
lower_green = np.array([35, 110, 106]) # 绿色阈值下界
higher_green = np.array([77, 255, 255]) # 绿色阈值上界
data_dir = '/home/美食/'for parent, dirs, files in os.walk(data_dir): # 遍历 data_dir 文件夹for file in files:cur_path = os.path.join(parent, file) # 当前遍历 data_dir 文件夹img = cv2.imread(cur_path)img_hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)mask_red = cv2.inRange(img_hsv, lower_red, higher_red) # 可以认为是过滤出红色部分,获得红色的掩膜mask_green =
这篇关于从数据集中挑选红色区域大于某一阈值的图片的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!