Skip to content

Instantly share code, notes, and snippets.

@david-littlefield

david-littlefield/dsfd_error_05_demo.py

Last active November 6, 2020 20:07
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save david-littlefield/f5114b2a55f71441bf7de7efb8676688 to your computer and use it in GitHub Desktop.
Save david-littlefield/f5114b2a55f71441bf7de7efb8676688 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
dsfd_error_05_demo.py
# Line 83
while detections[0, i, j, 0] >= thresh:
score = detections[0, i, j, 0]
score = score.cpu().numpy() # <----------
#label_name = labelmap[i-1]
pt = (detections[0, i, j, 1:]*scale).cpu().numpy()
coords = (pt[0], pt[1], pt[2], pt[3])
det.append([pt[0], pt[1], pt[2], pt[3], score])
j += 1
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment