|
import cv2
|
|
from pathlib import Path
|
|
import tqdm
|
|
import pickle
|
|
import os
|
|
|
|
import LFUtilities as lf
|
|
|
|
orb = cv2.ORB.create(lf.KEYPOINTS)
|
|
|
|
|
|
def extract(img_path):
|
|
img = lf.resize(lf.IMG_SIZE, cv2.imread(img_path))
|
|
kp, des = orb.detectAndCompute(img, mask=None)
|
|
return (kp, des)
|
|
|