github
/
darknet
mirror of https://github.com/AlexeyAB/darknet.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Another minor fix

Browse Source
pull/4477/head
AlexeyAB 6 years ago
parent 9bb7455a0e
commit fc663f6efe
2 changed files with 10 additions and 6 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 8
      build/darknet/x64/darknet_video.py
  2. 8
      darknet_video.py

8
build/darknet/x64/darknet_video.py
Unescape View File

@ -25,12 +25,12 @@ def cvDrawBoxes(detections, img):
float(x), float(y), float(w), float(h)) float(x), float(y), float(w), float(h))
pt1 = (xmin, ymin) pt1 = (xmin, ymin)
pt2 = (xmax, ymax) pt2 = (xmax, ymax)
cv2.rectangle(img, pt1, pt2, (0, 255, 0), 2) cv2.rectangle(img, pt1, pt2, (0, 255, 0), 1)
cv2.putText(img, cv2.putText(img,
detection[0].decode() + detection[0].decode() +
" [" + str(round(detection[1] * 100, 2)) + "]", " [" + str(round(detection[1] * 100, 2)) + "]",
(pt1[0], pt1[1] + 20), cv2.FONT_HERSHEY_SIMPLEX, 1, (pt1[0], pt1[1] - 5), cv2.FONT_HERSHEY_SIMPLEX, 0.5,
[0, 255, 0], 4) [0, 255, 0], 2)
return img return img
@ -106,6 +106,8 @@ def YOLO():
image = cvDrawBoxes(detections, frame_resized) image = cvDrawBoxes(detections, frame_resized)
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB) image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
print(1/(time.time()-prev_time)) print(1/(time.time()-prev_time))
cv2.imshow('Demo', image)
cv2.waitKey(3)
cap.release() cap.release()
out.release() out.release()

8
darknet_video.py
Unescape View File

@ -25,12 +25,12 @@ def cvDrawBoxes(detections, img):
float(x), float(y), float(w), float(h)) float(x), float(y), float(w), float(h))
pt1 = (xmin, ymin) pt1 = (xmin, ymin)
pt2 = (xmax, ymax) pt2 = (xmax, ymax)
cv2.rectangle(img, pt1, pt2, (0, 255, 0), 2) cv2.rectangle(img, pt1, pt2, (0, 255, 0), 1)
cv2.putText(img, cv2.putText(img,
detection[0].decode() + detection[0].decode() +
" [" + str(round(detection[1] * 100, 2)) + "]", " [" + str(round(detection[1] * 100, 2)) + "]",
(pt1[0], pt1[1] + 20), cv2.FONT_HERSHEY_SIMPLEX, 1, (pt1[0], pt1[1] - 5), cv2.FONT_HERSHEY_SIMPLEX, 0.5,
[0, 255, 0], 4) [0, 255, 0], 2)
return img return img
@ -106,6 +106,8 @@ def YOLO():
image = cvDrawBoxes(detections, frame_resized) image = cvDrawBoxes(detections, frame_resized)
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB) image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
print(1/(time.time()-prev_time)) print(1/(time.time()-prev_time))
cv2.imshow('Demo', image)
cv2.waitKey(3)
cap.release() cap.release()
out.release() out.release()

Write Preview
Loading…
Cancel
Save