if not ret_val:
print ('ERROR: no se pudo abrir la camara, saliendo')
exit()
imgFile3 = cv2.cvtColor(imgFile2, cv2.COLOR_BGR2RGB)
#imgFile2 = cv2.imread("../data/eagle.jpg")
tama=imgFile2.shape
imgImported=make_image(tama[1],tama[0],tama[2])
imgFileptr,cv_img=get_iplimage_ptr(imgFile3)
ipl_in2_image(imgFileptr,imgImported)
rgbgr_image(imgImported)
lineaDeConteo=[]
for cc in range(lineasDeConteo):
lineaDeConteo.append(lc.selectLine(imgFile2,ownString='Selecciona la linea de conteo #' +str(cc+1),filename=fn,linecount=cc+1))
sleep(1)
for fn in filelist:
#fn = easygui.fileopenbox(default="/media/francisco/SiliconPowerArmor/SDM/",filetypes = ['*.avi','*.mp4'])
cam = cv2.VideoCapture(fn)
ruta,ext=os.path.splitext(fn)
archsal=ruta+'.csv'
frames=0
ret_val, imgFile2 = cam.read()
frames+=1
if not ret_val:
print ('ERROR: no se pudo abrir la camara, saliendo')
exit()
imgFile3 = cv2.cvtColor(imgFile2, cv2.COLOR_BGR2RGB)
#imgFile2 = cv2.imread("../data/eagle.jpg")
tama = imgFile2.shape
imgImported = make_image(tama[1], tama[0], tama[2])
imgFileptr, cv_img = get_iplimage_ptr(imgFile3)
ipl_in2_image(imgFileptr, imgImported)
rgbgr_image(imgImported)
lineaDeConteo = []
for cc in range(lineasDeConteo):
lineaDeConteo.append(
lc.selectLine(imgFile2,
ownString='Selecciona la linea de deteccion #' +
str(cc + 1),
filename=folder + "/deteccion.jpg",
linecount=cc + 1))
sleep(1)
regionCebra = []
for rc in range(regionesZebra):
regionCebra.append(
lc.selectRect(imgFile2,
ownString='Selecciona la region de deteccion #' +
str(cc + 1),
filename=folder + "/deteccion.jpg",
linecount=cc + 1))
sleep(1)
pp = picoypla()
imgFile3 = cv2.cvtColor(imgFile2, cv2.COLOR_BGR2RGB)
#imgFile2 = cv2.imread("../data/eagle.jpg")
tama = imgFile2.shape
imgImported = make_image(tama[1], tama[0], tama[2])
imgFileptr, cv_img = get_iplimage_ptr(imgFile3)
ipl_in2_image(imgFileptr, imgImported)
rgbgr_image(imgImported)
lineaDeConteo = []
for cc in range(lineasDeConteo):
if not CONFIGURARPORDEFECTO:
lineaDeConteo.append(
lc.selectLine(imgFile2,
ownString='Selecciona la linea de deteccion #' +
str(cc + 1),
filename=folder + "/deteccion.jpg",
linecount=cc + 1))
else:
print("Seleccionando linea por defecto")
lineaDeConteo.append(
lc.selectLine(imgFile2,
ownString='Selecciona la linea de deteccion #' +
str(cc + 1),
filename=folder + "/deteccion.jpg",
linecount=cc + 1,
DefaultPoints=True,
pt1=ppt1,
pt2=ppt2))
sleep(1)
"""
print('ERROR: no se pudo abrir la camara, saliendo')
exit()
title = "Cuantas lineas de conteo?"
msg = "Seleccione el numero de lineas de conteo que quiere poner, se recomiendan maximo 6 lineas de conteo"
choices = ["1", "2", "3", "4", "5", "6"]
choice = easygui.choicebox(msg, title, choices)
type(choice)
lineasDeConteo = int(choice)
print "usted ha seleccionado ", lineasDeConteo, " lineas de conteo"
contadores = []
for cc in range(lineasDeConteo):
lineaDeConteo = lc.selectLine(imgFile2,
ownString='Selecciona la linea de conteo #' +
str(cc + 1),
filename=archsal,
linecount=cc + 1)
archivosAEnviar.append(lineaDeConteo.archivosalidajpg)
contadores.append(
lc.counter(lineaDeConteo.pt1,
lineaDeConteo.pt2,
filename=archsal,
linecount=cc + 1,
fps=20))
sleep(0.5)
for cnt in contadores:
cnt.saveLine()
archivosAEnviar.append(cnt.filename_output_line)
archivosAEnviar.append(cnt.filename_output)
imgFileptr, cv_img = get_iplimage_ptr(imgFile3)
ipl_in2_image(imgFileptr, imgImported)
rgbgr_image(imgImported)
track = tr.tracking(verbose=charlador, mindist=mindist, framesttl=framesttl)
lineasDeConteo = 1 # only one couting line
print("Only ", lineasDeConteo, " line for spatio-temporal accumulation")
contadores = []
for cc in range(lineasDeConteo):
sleep(1)
lineaDeConteo = lc.selectLine(
imgFile2,
ownString='Select the line for the spatio-temporal accumulation #' +
str(cc + 1),
filename=archsal,
linecount=cc + 1)
# parameters of each line of counting
sleep(1)
contadores.append(
lc.counter(lineaDeConteo.pt1,
lineaDeConteo.pt2,
filename=archsal,
linecount=cc + 1,
fps=20))
count = 0 #counter for the length of the image
len_SPT = 416 # length of the STI could me more
number_SPT = 1 #number of the STI processed
#save_image(imgImported,"eagle_detect")
r = detect_img(net, meta, imgImported)
print r
for i in range(len(r)):
w=int(r[i][2][2])
h=int(r[i][2][3])
x=int(r[i][2][0])-w/2
y=int(r[i][2][1])-h/2
cv2.rectangle(imgFile2, (x,y), (x+w,y+h), (255,255,0), thickness=1, lineType=8, shift=0)
cv2.imshow('Video', imgFile2)
cv2.waitKey(10)
"""
track = tr.tracking(verbose=charlador, mindist=mindist,
framesttl=framesttl) #verbose=False,mindist=100
lineaDeConteo = lc.selectLine(imgFile2,
ownString='Selecciona la linea de conteo',
filename=archsal,
linecount=1)
lineaDeConteo2 = lc.selectLine(imgFile2,
ownString='Selecciona la linea de conteo',
filename=archsal,
linecount=2)
contar = lc.counter(lineaDeConteo.pt1,
lineaDeConteo.pt2,
filename=archsal,
linecount=1,
fps=20)
contar2 = lc.counter(lineaDeConteo2.pt1,
lineaDeConteo2.pt2,
filename=archsal,
