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,