mirror of
https://github.com/serengil/deepface.git
synced 2025-06-08 12:35:22 +00:00
some more check for find test
This commit is contained in:
parent
0eb1515e11
commit
d35833e4e1
@ -7,19 +7,42 @@ logger = Logger("tests/test_find.py")
|
||||
|
||||
|
||||
def test_find_with_exact_path():
|
||||
dfs = DeepFace.find(img_path="dataset/img1.jpg", db_path="dataset", silent=True)
|
||||
img_path = "dataset/img1.jpg"
|
||||
dfs = DeepFace.find(img_path=img_path, db_path="dataset", silent=True)
|
||||
assert len(dfs) > 0
|
||||
for df in dfs:
|
||||
assert isinstance(df, pd.DataFrame)
|
||||
|
||||
# one is img1.jpg itself
|
||||
identity_df = df[df["identity"] == img_path]
|
||||
assert identity_df.shape[0] > 0
|
||||
|
||||
# validate reproducability
|
||||
assert identity_df["VGG-Face_cosine"].values[0] == 0
|
||||
|
||||
df = df[df["identity"] != img_path]
|
||||
logger.debug(df.head())
|
||||
assert df.shape[0] > 0
|
||||
logger.info("✅ test find for exact path done")
|
||||
|
||||
|
||||
def test_find_with_array_input():
|
||||
img1 = cv2.imread("dataset/img1.jpg")
|
||||
img_path = "dataset/img1.jpg"
|
||||
img1 = cv2.imread(img_path)
|
||||
dfs = DeepFace.find(img1, db_path="dataset", silent=True)
|
||||
|
||||
assert len(dfs) > 0
|
||||
for df in dfs:
|
||||
assert isinstance(df, pd.DataFrame)
|
||||
|
||||
# one is img1.jpg itself
|
||||
identity_df = df[df["identity"] == img_path]
|
||||
assert identity_df.shape[0] > 0
|
||||
|
||||
# validate reproducability
|
||||
assert identity_df["VGG-Face_cosine"].values[0] == 0
|
||||
|
||||
|
||||
df = df[df["identity"] != img_path]
|
||||
logger.debug(df.head())
|
||||
assert df.shape[0] > 0
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user