Skip to content

Instantly share code, notes, and snippets.

@arunm8489

arunm8489/36

Created June 5, 2020 05:08
Show Gist options
  • Save arunm8489/11afa19e81bb1e35814e072dc3b1726c to your computer and use it in GitHub Desktop.
Save arunm8489/11afa19e81bb1e35814e072dc3b1726c to your computer and use it in GitHub Desktop.
Download ZIP
Raw
36
# sort them based on probability #getting index
conf_sort_index = torch.sort(image_pred_class[:,4], descending = True )[1]
image_pred_class = image_pred_class[conf_sort_index]
idx = image_pred_class.size(0)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment