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
a1d94d40
Commit
a1d94d40
authored
Sep 11, 2020
by
赵威
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
rerun
parent
f8c636df
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
17 additions
and
15 deletions
+17
-15
diary_cover_similarity.py
src/diary_cover_similarity.py
+17
-15
No files found.
src/diary_cover_similarity.py
View file @
a1d94d40
...
...
@@ -55,21 +55,23 @@ def save_diary_image_info(save_file, face_to_vec_f):
}
count
=
0
with
open
(
save_file
,
"w"
)
as
f
:
res_dict
=
es_query
(
"diary"
,
q
,
0
,
500000
)
for
item
in
res_dict
[
"hits"
][
"hits"
]:
diary_id
=
item
[
"_source"
][
"id"
]
# before_cover_url = item["_source"]["before_cover_url"] + "-w"
after_cover_url
=
item
[
"_source"
][
"after_cover_url"
]
+
"-w"
img
=
url_to_ndarray
(
after_cover_url
)
if
img
.
any
():
count
+=
1
print
(
"count: "
+
str
(
count
))
faces
=
face_to_vec_f
(
img
)
for
face
in
faces
:
line
=
str
(
diary_id
)
+
"
\t
"
+
face
[
"feature"
]
+
"
\n
"
# print(line)
f
.
write
(
line
)
steps
=
[(
0
,
200000
),
(
200000
,
400000
),
(
400000
,
600000
)]
for
(
start
,
stop
)
in
steps
:
res_dict
=
es_query
(
"diary"
,
q
,
start
,
stop
)
for
item
in
res_dict
[
"hits"
][
"hits"
]:
diary_id
=
item
[
"_source"
][
"id"
]
# before_cover_url = item["_source"]["before_cover_url"] + "-w"
after_cover_url
=
item
[
"_source"
][
"after_cover_url"
]
+
"-w"
img
=
url_to_ndarray
(
after_cover_url
)
if
img
.
any
():
count
+=
1
print
(
"count: "
+
str
(
count
))
faces
=
face_to_vec_f
(
img
)
for
face
in
faces
:
line
=
str
(
diary_id
)
+
"
\t
"
+
face
[
"feature"
]
+
"
\n
"
# print(line)
f
.
write
(
line
)
def
save_faiss_index
(
load_file
,
save_path
):
...
...
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