28 lines
958 B
Python
28 lines
958 B
Python
from util import embed_scores as ES
|
|
|
|
query_vector = ES.get_query_vector('A cat grooming their tail')
|
|
c_dir = '/srv/ftp_tcc/leopards1/2026/02/28'
|
|
|
|
vpath = '/srv/ftp_tcc/leopards2/2026/02/28/Leopards2_00_20260228210054.mp4'
|
|
from CommonCode.video_meta import FTPVideo
|
|
#vec_rep = ES.get_vector_representation('/srv/ftp_tcc/leopards2/2026/02/28')
|
|
#if isinstance(vec_rep['embeds'], dict):
|
|
# vec_rep['embeds'] = [y for x,y in vec_rep['embeds'].items()]
|
|
# %%
|
|
vec_rep = ES.get_vector_representation(c_dir)
|
|
vec_rep['idces'].shape
|
|
sum([x.shape[-2] for _,x in vec_rep['embeds'].items()])
|
|
# %%
|
|
#ou = ES.calculate_embedding_score_in_folder(
|
|
ou = ES.calculate_embedding_score_in_folder(c_dir, threshold=0.04, query = 'A cat grooming')
|
|
# %%
|
|
vec_rep, query_scores = ES.get_scores_embedding_c_dir(c_dir, query_vector.squeeze())
|
|
|
|
# results = ES.get_clip_scores(video_path, prompt)
|
|
# %%
|
|
|
|
|
|
import pickle
|
|
with open('/home/thebears/crap.p','rb') as ff:
|
|
ou = pickle.load(ff)
|