features.py 文件源码

def draw_features(img, features):
    """Draw features"""
    # Convert to OpenCV KeyPoints
    cv_kps = []
    for f in features:
        cv_kps.append(cv2.KeyPoint(f.pt[0], f.pt[1], f.size))

    # Draw keypoints
    img = cv2.drawKeypoints(img, cv_kps, None, color=(0, 255, 0))
    return img