34 lines
789 B
Python
34 lines
789 B
Python
|
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
|