Skip to content
Snippets Groups Projects
Commit 97d68603 authored by Schneider Leo's avatar Schneider Leo
Browse files

rm : data prediction pred

parent affef4b5
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 0 additions and 2 deletions
image_ref/main.py
+ 0
− 2
View file @ 97d68603
...@@ -238,8 +238,6 @@ def make_prediction_duo(model, data, f_name, f_name2): ...@@ -238,8 +238,6 @@ def make_prediction_duo(model, data, f_name, f_name2):
label = label.cuda() label = label.cuda()
output = model(imaer, imana, img_ref) output = model(imaer, imana, img_ref)
confidence = soft_max(output) confidence = soft_max(output)
print(label)
print(confidence)
confidence_pred_list[specie].append(confidence[:, 0].data.cpu().numpy()) confidence_pred_list[specie].append(confidence[:, 0].data.cpu().numpy())
# Mono class output (only most postive paire) # Mono class output (only most postive paire)
output = torch.argmax(confidence[:, 0]) output = torch.argmax(confidence[:, 0])
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment