Initial commit

Divers/tutoriel36-2/affiche_video_label.py

@@ -0,0 +1,33 @@
+import cv2
+import numpy as np
+from numpy import genfromtxt
+import os
+import glob
+
+datasets="2DMOT2015Labels/train/"
+dataset="PETS09-S2L1"
+
+dir_images=datasets+"/"+dataset+"/img1/"
+fichier_label=datasets+"/"+dataset+"/gt/gt.txt"
+
+if not os.path.exists(fichier_label):
+    print("Le fichier de label n'existe pas ...", fichier)
+    quit()
+
+data=genfromtxt(fichier_label, delimiter=',')
+id_frame=0
+id_objet=0
+
+for image in glob.glob(dir_images+"*.jpg"):
+    frame=cv2.imread(image)
+
+    mask=data[:, 0]==id_frame
+    for d in data[mask, :]:
+        cv2.rectangle(frame, (int(d[2]), int(d[3])), (int(d[2]+d[4]), int(d[3]+d[5])), (0, 255, 0), 2)
+
+    cv2.imshow("frame", frame)
+
+    key=cv2.waitKey(70)&0xFF
+    if key==ord('q'):
+        quit()
+    id_frame+=1