(train_data, labels, path, lableType)
| 18 | return fv |
| 19 | |
| 20 | def get_data(train_data, labels, path, lableType): |
| 21 | for file_name in os.listdir(path): |
| 22 | img_dir = os.path.join(path, file_name) |
| 23 | img = cv.imread(img_dir) |
| 24 | hog_desc = get_hog_descriptor(img) |
| 25 | one_fv = np.zeros([len(hog_desc)], dtype=np.float32) |
| 26 | for i in range(len(hog_desc)): |
| 27 | one_fv[i] = hog_desc[i][0] |
| 28 | train_data.append(one_fv) |
| 29 | labels.append(lableType) |
| 30 | return train_data, labels |
| 31 | |
| 32 | def get_dataset(pdir, ndir): |
| 33 | train_data = [] |
no test coverage detected