Skip to content

Commit

Permalink
Update 10.evaluate_triplets.py
Browse files Browse the repository at this point in the history
  • Loading branch information
AKASH2907 authored Jun 13, 2022
1 parent ddbcf97 commit df34bad
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions 10.evaluate_triplets.py
Original file line number Diff line number Diff line change
Expand Up @@ -262,9 +262,8 @@ def create_base_network(image_input_shape, embedding_size):

# test_data = np.load("test_embs.npy")
test_label = np.load("test_labels.npy")
y_test_onehot = utils.to_categorical(test_label)

for i in videos[:]:
for i in videos:
cap = cv2.VideoCapture(i)
batches = []
mounting = 0
Expand Down Expand Up @@ -299,11 +298,11 @@ def create_base_network(image_input_shape, embedding_size):

pred_mean = np.mean(y_pred, axis=0)
probab_mean = np.mean(y_probabs, axis=0)
probab_mean = 1 - probab_mean
# probab_mean = 1 - probab_mean

y_probabilities +=[probab_mean]
# print(pred_mean)
if pred_mean>0.5:
if pred_mean<0.5:
y_predictions+=[0]
else:
y_predictions+=[1]
Expand Down

0 comments on commit df34bad

Please sign in to comment.