28 Star 126 Fork 24

cungudafa / hand-keras-yolo3-recognize

Clone or download
get_features.py
Copy Edit Web IDE Raw Normal History
cungudafa   sign 2020-08-08
# 从人脸图像文件中提取人脸特征存入 CSV# Features extraction from images and save into features_all.csv# return_128d_features()          获取某张图像的128D特征# compute_the_mean()              计算128D特征均值from cv2 import cv2 as cv2import osimport csvimport numpy as npimport timefrom pose_hand import getImgInfofrom yolo import YOLOfrom pose.coco import general_coco_model# ----------------------------------------------------------------------------------# 第一步 读取标签Y和图片路径X# ----------------------------------------------------------------------------------X = []  # 定义图像名称Y = []  # 定义图像分类类标# Z = [] #定义图像像素path = 'D:/myworkspace/dataset/My_test/dataset/test'for idx, labelname in enumerate(os.listdir(path)):    if ".txt" not in labelname:        f = os.path.join(path, labelname)        for i, imgname in enumerate(os.listdir(f)):            imgpath = os.path.join(f, imgname)            X.append(imgpath)            Y.append(labelname)X = np.array(X)Y = np.array(Y)# ----------------------------------------------------------------------------------# 第二步 识别infolist# ----------------------------------------------------------------------------------# cocomodelpath = "model/"start = time.time()pose_model = general_coco_model(modelpath)  # 1.加载模型print("[INFO]Pose Model loads time: ", time.time() - start)# yolostart = time.time()_yolo = YOLO()  # 1.加载模型print("[INFO]yolo Model loads time: ", time.time() - start)infolist = []for i in X:    hist = getImgInfo(i, pose_model, _yolo) # 识别    infolist.append(hist)# ----------------------------------------------------------------------------------# 第三步 存储信息docs/feature/features_all.csv# ----------------------------------------------------------------------------------# 路径存储到txtorb = open('D:/myworkspace/dataset/My_test/bagofwords/y_train.txt', 'w')for i, img_path in enumerate(X):    orb.write(img_path)    #orb.write('\n'+str(info)+str(infolist[i]))orb.close()# 特征存储到 csvwith open("docs/feature/features_all.csv", "w", newline="") as csvfile:    writer = csv.writer(csvfile)    for i in infolist:            writer.writerow(i)    print("所有录入手语特征数据存入 / Save all the features of sign registered into: docs/feature/features_all.csv")
Python
1
https://gitee.com/cungudafa/hand-keras-yolo3-recognize.git
git@gitee.com:cungudafa/hand-keras-yolo3-recognize.git
cungudafa
hand-keras-yolo3-recognize
hand-keras-yolo3-recognize
master

Search