图片特征HOG/LBP/ORB存为npy文件

文章目录

  • 前言
  • 一、HOG特征存npy文件
  • 二、LBP特征存npy文件
  • 三、ORB特征存npy文件
  • 总结


前言

例如需要分类器KNN、SVM等对图片特征进行分类,在这之前把图片特征转为npy文件。


一、HOG特征存npy文件

可以实现文件夹内的所有图片获取hog特征后存npy文件。

from skimage import feature as ft
import cv2
import matplotlib.pyplot as plt
import numpy as np

savepath = r"/PIC_F/A"#存到哪个位置
rootpath = r"PIC_F/hello/" #从哪个文件夹找到图片
imgfiles = os.listdir(rootpath)#形成文件途径

pic_weidu=400
npdata=np.empty([0,pic_weidu])#先固定400维度 和图片大小相关

def  Tiqu(imgpath):
    frame=cv2.imread(imgpath)    
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    normalised_blocks,features = ft.hog(gray,orientations=6,pixels_per_cell=[8,8],cells_per_block=[2,2],visualize=True)
    features=np.array(features,dtype=np.float32)
    # print(features.shape)
    return features

for i in range(0, len(imgfiles)):
    path = os.path.join(rootpath, imgfiles[i])
    if os.path.isfile(path):
        if (imgfiles[i].endswith("jpg") or imgfiles[i].endswith("JPG")):
            picpath =os.path.join(rootpath, imgfiles[i])#找到每一张图片
            dst=Tiqu(picpath)
            npdata= np.append(npdata, dst, axis=0)
print("数据总的维度是%d %d" % npdata.shape)
np.save(r".\PIC_F\a\A.npy",npdata)

二、LBP特征存npy文件

from skimage.transform import rotate
from skimage.feature import local_binary_pattern
from skimage import data, io
from skimage.color import label2rgb
import skimage
import numpy as np
import matplotlib.pyplot as plt
from PIL import Image
import cv2

# settings for LBP
radius = 3
n_points = 8 * radius


savepath = r"/PIC_F/A"#存到哪个位置
rootpath = r"PIC_F/hello/" #从哪个文件夹找到图片
imgfiles = os.listdir(rootpath)#形成文件途径

pic_weidu=400
npdata=np.empty([0,pic_weidu])#先固定400维度

def  Tiqu(imgpath):
    frame=cv2.imread(imgpath)    
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    lbp = local_binary_pattern(image, n_points, radius)
    # print(lbp.shape)
    features=np.array(lbp,dtype=np.float32)
    # print(features.shape)
    return features

for i in range(0, len(imgfiles)):
    path = os.path.join(rootpath, imgfiles[i])
    if os.path.isfile(path):
        if (imgfiles[i].endswith("jpg") or imgfiles[i].endswith("JPG")):
            picpath =os.path.join(rootpath, imgfiles[i])#找到每一张图片
            dst=Tiqu(picpath)
            npdata= np.append(npdata, dst, axis=0)
print("数据总的维度是%d %d" % npdata.shape)
np.save(r".\PIC_F\a\B.npy",npdata)

三、ORB特征存npy文件

SIFT的特征也可以参考这个处理

import os
import cv2
import numpy as np

savepath = r"/PIC_F/A"#存到哪个位置
rootpath = r"PIC_F/hello/" #从哪个文件夹找到图片
imgfiles = os.listdir(rootpath)#形成文件途径

pic_weidu=32
npdata=np.empty([0,pic_weidu])#先固定32维度

Ex_weidu=315  #特征点统一一个维度 这个获取的每个图片特征点数量不一致,我们在分类器中不方便使用,所以统一一个维度
#不满足这个数量的特征就填充为0

def  Tiqu(imgpath):
    frame=cv2.imread(imgpath)    
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    orb=cv2.ORB_create()
    kp,des=orb.detectAndCompute(gray,None)
    features=np.array(des,dtype=np.float32)
    print("图片特征获取之前的维度%d" % len(kp))    
    fillnpdata=np.zeros([Ex_weidu-len(kp),pic_weidu])
    features=np.append(features, fillnpdata, axis=0)
    return features

for i in range(0, len(imgfiles)):
    path = os.path.join(rootpath, imgfiles[i])
    if os.path.isfile(path):
        if (imgfiles[i].endswith("jpg") or imgfiles[i].endswith("JPG")):
            picpath =os.path.join(rootpath, imgfiles[i])#找到每一张图片
            dst=Tiqu(picpath)
            npdata= np.append(npdata, dst, axis=0)
print("数据总的维度是%d %d" % npdata.shape)
np.save(r".\PIC_F\a\C.npy",npdata)

总结

备忘。

你可能感兴趣的:(OpenCV,python,算法)