Skip to content
Projects
Groups
Snippets
Help
Loading...
Sign in
Toggle navigation
S
strategy_embedding
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
rank
strategy_embedding
Commits
3bd1ac93
Commit
3bd1ac93
authored
Sep 14, 2020
by
赵威
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
get result
parent
f48e1b8f
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
8 additions
and
3 deletions
+8
-3
diary_cover_similarity.py
src/diary_cover_similarity.py
+8
-3
No files found.
src/diary_cover_similarity.py
View file @
3bd1ac93
import
json
import
os
import
random
import
time
import
dlib
...
...
@@ -192,6 +193,7 @@ def main():
# res = get_similar_diary_ids_by_face_features(a, faiss_index, face_to_vec_f)
# print(res)
res_dict
=
{}
with
open
(
diary_after_cover_vec_file
)
as
f
:
lines
=
f
.
readlines
()
print
(
"lines: "
+
str
(
len
(
lines
)))
...
...
@@ -202,9 +204,12 @@ def main():
id
=
tmp
[
0
]
feature
=
np
.
array
(
json
.
loads
(
tmp
[
1
]))
print
(
"{} {}"
.
format
(
count
,
id
))
res
=
get_similar_diary_ids_by_face_features
(
feature
,
faiss_index
,
face_to_vec_f
)
print
(
res
)
print
(
"========================="
)
tup_res
=
get_similar_diary_ids_by_face_features
(
feature
,
faiss_index
,
face_to_vec_f
)
if
tup_res
:
res_dict
[
id
]
=
tup_res
print
(
"done: "
+
str
(
len
(
res_dict
)))
key
=
random
.
choice
(
list
(
res_dict
.
keys
()))
print
(
res_dict
[
key
])
if
__name__
==
"__main__"
:
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment