def __main_function__():
# Import some global varialbes
global archivoDeVideo
global acaboDeIniciarNuevoCiclo
acaboDeIniciarNuevoCiclo = False
global tuveInfracciones
tuveInfracciones = False
# Creamos el reporte inicial
miReporte = MiReporte(levelLogging=logging.INFO,nombre=__name__) # Se crea por defecto con nombre de la fecha y hora actual
miReporte.info('Programa iniciado exitosamente con ingreso de senal video '+archivoDeVideo+entradaReal+' con semaforo '+semaforoSimuladoTexto+str(periodoDeSemaforo) +', corriendo a '+str(mifps)+' Frames por Segundo')
# Si no existe el directorio de reporte lo creo
if not os.path.exists(directorioDeReporte):
os.makedirs(directorioDeReporte)
# Vector de inicio:
# vector de inicio = [tiempo, periodo semaforo, cruce, giro, infraccion, otros]
vectorDeInicio = [[datetime.datetime.now(),0,0,0,0,0]]
if os.path.isfile(reporteDiario):
miReporte.info('Continuando reporte')
np.save(reporteDiario,np.append(np.load(reporteDiario),vectorDeInicio,0))
else:
miReporte.info('Creando reporte desde cero')
np.save(reporteDiario,vectorDeInicio)
# Is statements
if generarArchivosDebug:
miReporte.info('Generando Archivos de Debug')
else:
miReporte.info('Generando infracciones unicamente (No debug video)')
# If mostrar Imagenes
if mostrarImagen:
miReporte.info('Pantalla de muestra de funcionamiento en tiempo real encendida')
else:
miReporte.info('Pantalla de muestra de funcionamiento en tiempo real apagada')
# El directorio de reporte debe crearse al inicio del programa
# Variables de control:
numeroDeFrame = 0
maximoInfraccionesPorFrame = 20
#colores = np.random.randint(0,100,(maximoInfraccionesPorFrame,3))
# Cargando los parametros de instalacion:
# El archivo de video debe tener como minimo 5 caracteres para estar trnajando en modo simulado, de lo contrario estamos trabajando en modo real
if len(archivoDeVideo) > 4:
archivoParametrosACargar = archivoDeVideo[:-4]+'.npy'
else:
archivoParametrosACargar = 'datos.npy'
parametrosInstalacion = np.load(folderDeInstalacion+'/'+archivoParametrosACargar)
miReporte.info('Datos de Instalacion de: '+folderDeInstalacion+'/'+archivoParametrosACargar)
poligonoSemaforo = parametrosInstalacion[0]
verticesPartida = parametrosInstalacion[1]
verticesLlegada = parametrosInstalacion[2]
indicesSemaforo = obtenerIndicesSemaforo(np.array(poligonoSemaforo))
angulo = parametrosInstalacion[3]
poligonoEnAlta = parametrosInstalacion[4]
miReporte.info('Cargado exitosamente parametros de instalacion ')#+str(parametrosInstalacion))
# Arrancando camara
if len(archivoDeVideo) == 0:
conVideoGrabado = False # modo real
miCamara = cv2.VideoCapture(0)
miCamara.set(3,640)
miCamara.set(4,480)
time.sleep(1)
miReporte.info('Activada Exitosamente cámara en tiempo real')
else:
conVideoGrabado = True
try:
miCamara = cv2.VideoCapture(directorioDeVideos+'/'+archivoDeVideo)
time.sleep(1)
miReporte.info('Archivo de video cargado exitosamente: '+directorioDeVideos+'/'+archivoDeVideo)
except Exception as currentException:
miReporte.error('No se pudo cargar el video por '+str(currentException))
# Se captura la imagen de flujo inicial y se trabaja con la misma
ret, frameVideo = miCamara.read()
frameFlujo = cv2.resize(frameVideo,(320,240))
# Creación de objetos:
if os.uname()[1] == 'raspberrypi':
trabajoConPiCamara = True
else:
trabajoConPiCamara = False
miPoliciaReportando = PoliciaInfractor(frameFlujo,verticesPartida,verticesLlegada)
miGrabadora = GeneradorEvidencia(directorioDeReporte,mifps,False)
miFiltro = IRSwitch()
miAcetatoInformativo = Acetato()
miSemaforo = CreateSemaforo(periodoDeSemaforo)
miAcetatoInformativo.colocarPoligono(np.array(poligonoSemaforo)//2)
miAcetatoInformativo.colocarPoligono(np.array(verticesPartida))
miAcetatoInformativo.colocarPoligono(np.array(verticesLlegada))
miAcetatoInformativo.colocarPoligono(miPoliciaReportando.carrilValido)
# El historial sera una lista de la siguiente forma:
# {numeroFrame: {'frame':np.array((320,240)),'data':{"info"}}}
historial = {}
frame_number = 0
tiempoAuxiliar = time.time()
periodoDeMuestreo = 1.0/mifps
grupo = [0]
try:
while True:
# LEEMOS LA CAMARA DE FLUJO
if conVideoGrabado:
for i in range(videofps//mifps):
ret, frameVideo = miCamara.read()
else:
ret, frameVideo = miCamara.read()
pixeles = np.array([frameVideo[indicesSemaforo[0][1],indicesSemaforo[0][0]]])
#print('IndicesPixel: ',indicesSemaforo[0][0],indicesSemaforo[0][1])
#print('La longitud semaforo: ',len(indicesSemaforo),' inicial ',pixeles.shape)
#print('La longitud interna: ',len(indicesSemaforo[0]),' inicial ',pixeles.shape)
for indiceSemaforo in indicesSemaforo[1:]:
pixeles = np.append(pixeles,[frameVideo[indiceSemaforo[1],indiceSemaforo[0]]], axis=0)
#cv2.circle(frameVideo, (indiceSemaforo[0],indiceSemaforo[1]), 1, (100,100,100), -1)
#print('Pixeles: ',pixeles)
wtf = pixeles.reshape((24,8,3))
#cv2.imshow('Semaforo', cv2.resize(wtf, (240,320)))
#print('La longitud pixels: ',pixeles.shape)
senalSemaforo, semaforoLiteral, flanco, periodo = miSemaforo.obtenerColorEnSemaforo(pixeles)
frameFlujo = cv2.resize(frameVideo,(320,240))
if periodo != 0:
miReporte.info('SEMAFORO EN VERDE, EL PERIODO ES '+str(periodo))
miReporte.info('DateTime '+datetime.datetime.now().strftime('%Y%m%d_%H%M'))
cruce = miPoliciaReportando.estadoActual['cruzo']
giro = miPoliciaReportando.estadoActual['giro']
infraccion = miPoliciaReportando.estadoActual['infraccion']
otro = miPoliciaReportando.estadoActual['ruido']
vectorDeInicio = [[datetime.datetime.now(),periodo,cruce,giro,infraccion,otro]]
np.save(reporteDiario,np.append(np.load(reporteDiario),vectorDeInicio,0))
else:
pass
# Si tengo infracciones pendientes las evoluciono
velocidadEnBruto, velocidadFiltrada, pulsoVehiculos, momentumAEmplear = miPoliciaReportando.seguirImagen(frame_number,frameFlujo,colorSemaforo = senalSemaforo)
if senalSemaforo >= 1 : # Si estamos en rojo, realizamos una accion
if flanco == 1: # esto se inicia al principio de este estado
miReporte.info('SEMAFORO EN ROJO')
if senalSemaforo == 0: # Si estamos en verde realizamos otra accion
if flanco == -1: # Si estamos en verde y en flanco, primer verde, realizamos algo
miReporte.info('INICIANDO REPORTE DE: '+str(miPoliciaReportando.numeroInfraccionesConfirmadas())+' INFRACCIONES')
acaboDeIniciarNuevoCiclo = True
ultimoNumeroInfraccion = miPoliciaReportando.numeroInfraccionesConfirmadas()
if acaboDeIniciarNuevoCiclo:
if miPoliciaReportando.numeroInfraccionesConfirmadas() > 0:
tuveInfracciones = True
infraccionEnRevision = miPoliciaReportando.popInfraccion()
miGrabadora.generarReporteInfraccion(historial, infraccionEnRevision)
else:
if (tuveInfracciones):
if generarArchivosDebug:
miGrabadora.generarReporteInfraccion(historial, False,ultimoNumeroInfraccion)
tuveInfracciones = False
miPoliciaReportando.purgarInfraccionesRemanentes()
del historial
historial = {}
frame_number = 0
acaboDeIniciarNuevoCiclo = False
else:
#Si no hay infracciones a reportar me fijo el estado del filtro:
tiempoAhora = datetime.datetime.now().hour*60 + datetime.datetime.now().minute
if (tiempoAhora > amaneciendo) & (miFiltro.ultimoEstado != 'Filtro Activado'):
miFiltro.colocarFiltroIR()
if (tiempoAhora < amaneciendo) & (miFiltro.ultimoEstado != 'Filtro Desactivado'):
miFiltro.quitarFiltroIR()
# Draw frame number into image on top
for infraction in miPoliciaReportando.listaVehiculos:
for puntos in infraction['desplazamiento']:
puntosExtraidos = puntos.ravel().reshape(puntos.ravel().shape[0]//2,2)
miAcetatoInformativo.colocarObjeto(puntosExtraidos,infraction['estado'])
#for punto in puntosExtraidos:
# if infraction['estado'] == 'Confirmado':
# miAcetatoInformativo.colocarPunto(tuple(punto),0)
# else:
# miAcetatoInformativo.colocarPunto(tuple(punto),1)
#for puntoResguardo in miPoliciaReportando.obtenerLineasDeResguardo(False):
miAcetatoInformativo.colocarObjeto(miPoliciaReportando.obtenerLineasDeResguardo(True),'Referencia')
# Configs and displays for the MASK according to the semaforo
#miAcetatoInformativo.agregarTextoEn("I{}".format(miPoliciaReportando.infraccionesConfirmadas), 2)
miAcetatoInformativo.colorDeSemaforo(senalSemaforo)
historial[frame_number] = {'captura':frameFlujo.copy()}
frameFlujo = miAcetatoInformativo.aplicarAFrame(frameFlujo)
if mostrarImagen:
#cv2.imshow('Visual', miAcetatoInformativo.aplicarAFrame(frameFlujo)[120:239,60:360])
cv2.imshow('Visual',frameFlujo)
if generarArchivosDebug:
historial[frame_number]['frame'] = frameFlujo.copy()
else:
historial[frame_number]['frame'] = historial[frame_number]['captura']
historial[frame_number]['data'] = [velocidadEnBruto, velocidadFiltrada, pulsoVehiculos, momentumAEmplear]
miAcetatoInformativo.inicializar()
tiempoEjecucion = time.time() - tiempoAuxiliar
if tiempoEjecucion>periodoDeMuestreo:
miReporte.warning('\t[f{}'.format(frame_number)+']'+' Periodo Excedido {0:2f}'.format(tiempoEjecucion)+ '[s]')
#sys.stdout.write("\033[F")
while time.time() - tiempoAuxiliar < periodoDeMuestreo:
True
tiempoAuxiliar = time.time()
porcentajeDeMemoria = psutil.virtual_memory()[2]
if (porcentajeDeMemoria > 80)&(os.uname()[1] == 'raspberrypi'):
miReporte.info('Estado de Memoria: '+str(porcentajeDeMemoria)+'/100')
"""
if porcentajeDeMemoria > 96:
miReporte.warning('Alcanzado 96/100 de memoria, borrando todo e inicializando')
del historial
historial = {}
frame_number = 0
"""
frame_number += 1
if (frame_number >= topeEjecucion) &(topeEjecucion!=0):
miReporte.info('ABANDONANDO LA EJECUCION DE PROGRAMA por indice de auto acabado predeterminado')
miPoliciaReportando.apagarCamara()
break
if senalSemaforo == -2:
miReporte.critical('ABANDONANDO LA EJECUCION DE PROGRAMA El semaforo ya no obtuvo señal, necesito recalibrar, abandonando la ejecución del programa')
break
ch = 0xFF & cv2.waitKey(5)
if ch == ord('q'):
miReporte.info('ABANDONANDO LA EJECUCION DE PROGRAMA por salida manual')
miPoliciaReportando.apagarCamara()
break
if ch == ord('s'):
cv2.imwrite(datetime.datetime.now().strftime('%Y-%m-%d_%H:%M:%S')+'.jpg',frameFlujo)
except KeyboardInterrupt as e:
miReporte.info('Salida forzada')
miPoliciaReportando.apagarCamara()
def __main_function__():
# Import some global varialbes
global archivoDeVideo
global cambiosImportantes
global grabadoParalelo
cambiosImportantes = False
# Creamos el reporte inicial
miReporte = MiReporte(
levelLogging=logging.INFO, nombre=__name__
) # Se crea por defecto con nombre de la fecha y hora actual
miReporte.info(
'Programa iniciado exitosamente con ingreso de senal video ' +
archivoDeVideo + entradaReal + ' con semaforo ' +
semaforoSimuladoTexto + str(periodoDeSemaforo) + ', corriendo a ' +
str(mifps) + ' Frames por Segundo')
# Si no existe el directorio de reporte lo creo
if not os.path.exists(directorioDeReporte):
os.makedirs(directorioDeReporte)
# Is statements
if generarArchivosDebug:
miReporte.info('Generando Archivos de Debug')
else:
miReporte.info('Generando infracciones unicamente (No debug video)')
# If mostrar Imagenes
if mostrarImagen:
miReporte.info(
'Pantalla de muestra de funcionamiento en tiempo real encendida')
else:
miReporte.info(
'Pantalla de muestra de funcionamiento en tiempo real apagada')
# El directorio de reporte debe crearse al inicio del programa
# Variables de control:
numeroDeFrame = 0
maximoInfraccionesPorFrame = 20
#colores = np.random.randint(0,100,(maximoInfraccionesPorFrame,3))
# Cargando los parametros de instalacion:
# El archivo de video debe tener como minimo 5 caracteres para estar trnajando en modo simulado, de lo contrario estamos trabajando en modo real
if len(archivoDeVideo) > 4:
archivoParametrosACargar = archivoDeVideo[:-4] + '.npy'
else:
archivoParametrosACargar = 'puntos.npy'
parametrosInstalacion = np.load(folderDeInstalacion + '/' +
archivoParametrosACargar)
miReporte.info('Datos de Instalacion de: ' + folderDeInstalacion + '/' +
archivoParametrosACargar)
poligonoSemaforo = parametrosInstalacion[0]
verticesPartida = parametrosInstalacion[1]
verticesLlegada = parametrosInstalacion[2]
indicesSemaforo = obtenerIndicesSemaforo(np.array(poligonoSemaforo))
angulo = parametrosInstalacion[3]
poligonoEnAlta = parametrosInstalacion[4]
miReporte.info('Cargado exitosamente parametros de instalacion: ' +
str(parametrosInstalacion))
# Arrancando camara
if len(archivoDeVideo) == 0:
conVideoGrabado = False # modo real
miCamara = cv2.VideoCapture(1)
miCamara.set(3, 640)
miCamara.set(4, 480)
time.sleep(1)
miReporte.info('Activada Exitosamente cámara en tiempo real')
else:
conVideoGrabado = True
try:
miCamara = cv2.VideoCapture(directorioDeVideos + '/' +
archivoDeVideo)
time.sleep(1)
miReporte.info('Archivo de video cargado exitosamente: ' +
directorioDeVideos + '/' + archivoDeVideo)
except Exception as currentException:
miReporte.error('No se pudo cargar el video por ' +
str(currentException))
# Se captura la imagen de flujo inicial y se trabaja con la misma
ret, frameVideo = miCamara.read()
frameFlujo = cv2.resize(frameVideo, (320, 240))
# Creación de objetos:
miPoliciaReportando = PoliciaInfractor(frameFlujo, verticesPartida,
verticesLlegada, False)
miPoliciaReportando.establecerRegionInteresAlta(poligonoEnAlta)
miGrabadora = GeneradorEvidencia(directorioDeReporte, mifps, False)
miFiltro = IRSwitch()
miAcetatoInformativo = Acetato()
miSemaforo = CreateSemaforo(periodoDeSemaforo)
miAcetatoInformativo.colocarPoligono(np.array(poligonoSemaforo) // 2)
miAcetatoInformativo.colocarPoligono(np.array(verticesPartida))
miAcetatoInformativo.colocarPoligono(np.array(verticesLlegada))
informacionTotal = {}
frame_number = 0
tiempoAuxiliar = time.time()
periodoDeMuestreo = 1.0 / mifps
grupo = [0]
frameAlta = np.empty((320, 240, 3))
capturaEnAlta = False
imagenesEnAlta = []
while True:
# LEEMOS LA CAMARA DE FLUJO
if conVideoGrabado:
for i in range(videofps // mifps):
ret, frameVideo = miCamara.read()
else:
if capturaEnAlta:
ret, frameAlta = miCamara.read()
imagenesEnAlta.append([
miPoliciaReportando.ultimaCarpetaGuardado + '/imagen.jpg',
frameAlta
])
frameVideo = cv2.resize(frameAlta, (320, 240))
capturaEnAlta = False
miCamara.set(3, 640)
miCamara.set(4, 480)
else:
ret, frameVideo = miCamara.read()
pixeles = np.array(
[frameVideo[indicesSemaforo[0][1], indicesSemaforo[0][0]]])
#print('IndicesPixel: ',indicesSemaforo[0][0],indicesSemaforo[0][1])
#print('La longitud semaforo: ',len(indicesSemaforo),' inicial ',pixeles.shape)
#print('La longitud interna: ',len(indicesSemaforo[0]),' inicial ',pixeles.shape)
for indiceSemaforo in indicesSemaforo[1:]:
pixeles = np.append(
pixeles, [frameVideo[indiceSemaforo[1], indiceSemaforo[0]]],
axis=0)
#print('>>> ',pixeles.shape,' in ',indiceSemaforo)
#cv2.circle(frameVideo, (indiceSemaforo[0],indiceSemaforo[1]), 1, (100,100,100), -1)
#print('Pixeles: ',pixeles)
#wtf = pixeles.reshape((24,8,3))
#cv2.imshow('Semaforo', cv2.resize(wtf, (240,320)))
#print('La longitud pixels: ',pixeles.shape)
senalSemaforo, semaforoLiteral, flanco, periodo = miSemaforo.obtenerColorEnSemaforo(
pixeles)
frameFlujo = cv2.resize(frameVideo, (320, 240))
if periodo != 0:
miReporte.info('SEMAFORO EN VERDE, EL PERIODO ES ' + str(periodo))
else:
pass
# Si tengo infracciones pendientes las evoluciono
cambiosImportantes, ondaFiltrada, frenteAutomovil, flujoTotal = miPoliciaReportando.seguirImagen(
frame_number, frameFlujo, colorSemaforo=senalSemaforo)
if senalSemaforo >= 1: # Si estamos en rojo, realizamos una accion
if frenteAutomovil == 1:
miCamara.set(3, 3280)
miCamara.set(4, 2464)
capturaEnAlta = True
if flanco == 1: # esto se inicia al principio de este estado
miReporte.info('SEMAFORO EN ROJO')
miPoliciaReportando.inicializarAgente()
del informacionTotal
informacionTotal = {}
frame_number = 0
else:
pass
else:
pass
if senalSemaforo == 0: # Si estamos en verde realizamos otra accion
if flanco == -1: # Si estamos en verde y en flanco, primer verde, realizamos algo
miReporte.info(
'Infracciones: ' +
str(miPoliciaReportando.numeroInfraccionesConfirmadas()))
if generarArchivosDebug:
miGrabadora.generarReporteInfraccion(
informacionTotal, False,
miPoliciaReportando.numeroInfraccionesConfirmadas())
miPoliciaReportando.purgeInfractions()
if miPoliciaReportando.numeroInfraccionesConfirmadas() > 0:
infraccionEnRevision = miPoliciaReportando.popInfraccion()
miGrabadora.generarReporteInfraccion(informacionTotal,
infraccionEnRevision)
if miPoliciaReportando.numeroInfraccionesConfirmadas == 0:
for imagen in imagenesEnAlta:
print('>>>>> GUARDANDO COMO: ', imagen[0], ' shape ',
imagen[1].shape)
cv2.imwrite(imagen[0], imagen[1])
else:
#Si no hay infracciones a reportar me fijo el estado del filtro:
tiempoAhora = datetime.datetime.now(
).hour * 60 + datetime.datetime.now().minute
if (tiempoAhora > amaneciendo) & (miFiltro.ultimoEstado !=
'Filtro Activado'):
miFiltro.colocarFiltroIR()
if (tiempoAhora < amaneciendo) & (miFiltro.ultimoEstado !=
'Filtro Desactivado'):
miFiltro.quitarFiltroIR()
pass
# Draw frame number into image on top
for infraction in miPoliciaReportando.listaDeInfracciones:
for puntos in infraction['desplazamiento']:
puntosExtraidos = puntos.ravel().reshape(
puntos.ravel().shape[0] // 2, 2)
for punto in puntosExtraidos:
if infraction['estado'] == 'Confirmado':
miAcetatoInformativo.colocarPunto(tuple(punto), 0)
else:
miAcetatoInformativo.colocarPunto(tuple(punto), 1)
# Configs and displays for the MASK according to the semaforo
#miAcetatoInformativo.agregarTextoEn("I{}".format(miPoliciaReportando.infraccionesConfirmadas), 2)
miAcetatoInformativo.colorDeSemaforo(senalSemaforo)
frameFlujo = miAcetatoInformativo.aplicarAFrame(frameFlujo)
if mostrarImagen:
#cv2.imshow('Visual', miAcetatoInformativo.aplicarAFrame(frameFlujo)[120:239,60:360])
cv2.imshow('Visual', frameFlujo)
informacionTotal[frame_number] = frameFlujo.copy()
miAcetatoInformativo.inicializar()
tiempoEjecucion = time.time() - tiempoAuxiliar
if tiempoEjecucion > periodoDeMuestreo:
miReporte.warning('Tiempo Afuera {0:2f}'.format(tiempoEjecucion) +
'[s] en frame {}'.format(frame_number))
#sys.stdout.write("\033[F")
while time.time() - tiempoAuxiliar < periodoDeMuestreo:
True
tiempoAuxiliar = time.time()
if cambiosImportantes:
miReporte.info(
'F{} Sema: '.format(frame_number) + semaforoLiteral + ' I: ' +
str(miPoliciaReportando.numeroInfraccionesConfirmadas()) +
'/' + str(miPoliciaReportando.numeroInfraccionesTotales()))
porcentajeDeMemoria = psutil.virtual_memory()[2]
if (porcentajeDeMemoria > 80) & (os.uname()[1] == 'raspberrypi'):
miReporte.info('Estado de Memoria: ' + str(porcentajeDeMemoria) +
'/100')
if porcentajeDeMemoria > 96:
miReporte.warning(
'Alcanzado 96/100 de memoria, borrando todo e inicializando')
del informacionTotal
informacionTotal = {}
frame_number = 0
frame_number += 1
if (frame_number >= topeEjecucion) & (topeEjecucion != 0):
miReporte.info(
'ABANDONANDO LA EJECUCION DE PROGRAMA por indice de auto acabado predeterminado'
)
break
if senalSemaforo == -2:
miReporte.critical(
'ABANDONANDO LA EJECUCION DE PROGRAMA El semaforo ya no obtuvo señal, necesito recalibrar, abandonando la ejecución del programa'
)
break
ch = 0xFF & cv2.waitKey(5)
if ch == ord('q'):
miReporte.info(
'ABANDONANDO LA EJECUCION DE PROGRAMA por salida manual')
break
if ch == ord('s'):
cv2.imwrite(
datetime.datetime.now().strftime('%Y-%m-%d_%H:%M:%S') + '.jpg',
frameFlujo)
def __main_function__():
# Import some global varialbes
global archivoDeVideo
global cambiosImportantes
cambiosImportantes = False
global numeroDeObjetos
numeroDeObjetos = 0
# Creamos el reporte inicial
miReporte = MiReporte(
levelLogging=logging.INFO, nombre=__name__
) # Se crea por defecto con nombre de la fecha y hora actual
miReporte.info(
'Programa iniciado exitosamente con ingreso de senal video ' +
archivoDeVideo + entradaReal + ' con semaforo ' +
semaforoSimuladoTexto + str(periodoDeSemaforo) + ', corriendo a ' +
str(mifps) + ' Frames por Segundo')
# Si no existe el directorio de reporte lo creo
if not os.path.exists(directorioDeReporte):
os.makedirs(directorioDeReporte)
# Is statements
if generarArchivosDebug:
miReporte.info('Generando Archivos de Debug')
else:
miReporte.info('Generando infracciones unicamente (No debug video)')
# If mostrar Imagenes
if mostrarImagen:
miReporte.info(
'Pantalla de muestra de funcionamiento en tiempo real encendida')
else:
miReporte.info(
'Pantalla de muestra de funcionamiento en tiempo real apagada')
# El directorio de reporte debe crearse al inicio del programa
# Variables de control:
numeroDeFrame = 0
maximoInfraccionesPorFrame = 20
#colores = np.random.randint(0,100,(maximoInfraccionesPorFrame,3))
# Cargando los parametros de instalacion:
# El archivo de video debe tener como minimo 5 caracteres para estar trnajando en modo simulado, de lo contrario estamos trabajando en modo real
if len(archivoDeVideo) > 4:
archivoParametrosACargar = archivoDeVideo[:-4] + '.npy'
else:
archivoParametrosACargar = 'datos.npy'
parametrosInstalacion = np.load(folderDeInstalacion + '/' +
archivoParametrosACargar)
miReporte.info('Datos de Instalacion de: ' + folderDeInstalacion + '/' +
archivoParametrosACargar)
poligonoSemaforo = parametrosInstalacion[0]
verticesPartida = parametrosInstalacion[1]
verticesLlegada = parametrosInstalacion[2]
angulo = parametrosInstalacion[3]
miReporte.info('Cargado exitosamente parametros de instalacion: ' +
str(parametrosInstalacion))
# Arrancando camara
if len(archivoDeVideo) == 0: # modo real
if os.uname()[1] == 'alvarohurtado-305V4A':
miCamara = VideoStream(src=1,
resolution=(640, 480),
poligono=poligonoSemaforo,
debug=saltarFrames,
fps=mifps,
periodo=periodoDeSemaforo,
gamma=gamma).start()
time.sleep(1)
else:
miCamara = VideoStream(src=0,
resolution=(3296, 2512),
poligono=poligonoSemaforo,
debug=saltarFrames,
fps=mifps,
periodo=periodoDeSemaforo,
gamma=gamma).start()
#miCamara = VideoStream(src = 0, resolution = (1920,1080),poligono = poligonoSemaforo, debug = saltarFrames,fps = mifps, periodo = periodoDeSemaforo, gamma = gamma).start()
#miCamara = VideoStream(src = 0, resolution = (1280,960),poligono = poligonoSemaforo, debug = saltarFrames,fps = mifps, periodo = periodoDeSemaforo, gamma = gamma).start()
time.sleep(1)
miReporte.info('Activada Exitosamente cámara en tiempo real')
else:
try:
miCamara = VideoStream(src=directorioDeVideos + '/' +
archivoDeVideo,
resolution=(640, 480),
poligono=poligonoSemaforo,
debug=saltarFrames,
fps=mifps,
periodo=periodoDeSemaforo,
gamma=gamma).start()
time.sleep(1)
miReporte.info('Archivo de video cargado exitosamente: ' +
directorioDeVideos + '/' + archivoDeVideo)
except Exception as currentException:
miReporte.error('No se pudo cargar el video por ' +
str(currentException))
# Se captura la imagen de flujo inicial y se trabaja con la misma
informacion = miCamara.read()
# Creación de objetos:
miPoliciaReportando = PoliciaInfractor(informacion['frame'],
verticesPartida, verticesLlegada)
miGrabadora = GeneradorEvidencia(directorioDeReporte, mifps,
guardarRecortados)
miFiltro = IRSwitch()
miAcetatoInformativo = Acetato()
miAcetatoInformativo.colocarPoligono(np.array(poligonoSemaforo) // 2)
miAcetatoInformativo.colocarPoligono(np.array(verticesPartida))
miAcetatoInformativo.colocarPoligono(np.array(verticesLlegada))
fps = FPS().start()
informacionTotal = {}
frame_number = 0
tiempoAuxiliar = time.time()
periodoDeMuestreo = 1.0 / mifps
grupo = [0]
while True:
# LEEMOS LA CAMARA DE FLUJO
# Ways to access to the information
# information['frame'] ; numpy array containing the low res frame
# information['semaforo'] ; list like [self.senalColor, self.colorLiteral, self.flancoSemaforo, self.totalperiodo]
# information['recortados'] ; list like of tuples representing listaderecortados from hd frame [(_numpy arrays_)n+1]
# information['rectangulos'] ; list like of tuples representing listaderectangulos and centroids in frame [((x,y,h,w),(p1,p2))n+1]
# n+1 ; represent the 1 by 1 correspndencia de los rectangulos encontrados y imagenes recortadas
informacion = miCamara.read() # Ways to access
# Asign number rfame to the information from miCamara.read()
#informacion['index'] = frame_number
informacionTotal[frame_number] = informacion.copy(
) #<------ ese .copy() faltaba
print('Out: ', informacion['semaforo'][2])
# Si forzamos por entrada o si estamos en verde botamos la información de los rectangulos:
if (guardarRecortados == False) | (
informacionTotal[frame_number]['semaforo'][0] == 0):
del informacionTotal[frame_number]['recortados']
informacionTotal[frame_number]['recortados'] = {}
# Se reporta el periodo del semaforo si es necesario:
if informacion['semaforo'][3] != 0:
miReporte.info('SEMAFORO EN VERDE, EL PERIODO ES ' +
str(informacion['semaforo'][3]))
else:
pass
# Si tengo infracciones pendientes las evoluciono
if informacion['semaforo'][
0] >= 1: # Si estamos en rojo, realizamos una accion
if informacion['semaforo'][
2] == 1: # esto se inicia al principio de este estado
miReporte.info('SEMAFORO EN ROJO')
miPoliciaReportando.inicializarAgente()
del informacionTotal
informacionTotal = {}
frame_number = 0
else:
pass
cambiosImportantes = miPoliciaReportando.seguirObjeto(
frame_number, informacion)
else:
pass
if informacion['semaforo'][
0] == 0: # Si estamos en verde realizamos otra accion
if informacion['semaforo'][
2] == -1: # Si estamos en verde y en flanco, primer verde, realizamos algo
miReporte.info(
'Infracciones: ' +
str(miPoliciaReportando.numeroInfraccionesConfirmadas()))
if generarArchivosDebug:
miGrabadora.generarReporteInfraccion(
informacionTotal, False,
miPoliciaReportando.numeroInfraccionesConfirmadas())
if miPoliciaReportando.numeroInfraccionesConfirmadas() > 0:
infraccionEnRevision = miPoliciaReportando.popInfraccion()
miGrabadora.generarReporteInfraccion(informacionTotal,
infraccionEnRevision)
else:
#Si no hay infracciones a reportar me fijo el estado del filtro:
tiempoAhora = datetime.datetime.now(
).hour * 60 + datetime.datetime.now().minute
if (tiempoAhora > amaneciendo) & (miFiltro.ultimoEstado !=
'Filtro Activado'):
miFiltro.colocarFiltroIR()
if (tiempoAhora < amaneciendo) & (miFiltro.ultimoEstado !=
'Filtro Desactivado'):
miFiltro.quitarFiltroIR()
pass
# Draw frame number into image on top
for infraction in miPoliciaReportando.listaDeInfracciones:
for puntos in infraction['desplazamiento']:
puntosExtraidos = puntos.ravel().reshape(
puntos.ravel().shape[0] // 2, 2)
for punto in puntosExtraidos:
if infraction['estado'] == 'Confirmado':
miAcetatoInformativo.colocarPunto(tuple(punto), 0)
else:
miAcetatoInformativo.colocarPunto(tuple(punto), 1)
# Configs and displays for the MASK according to the semaforo
#miAcetatoInformativo.agregarTextoEn("I{}".format(miPoliciaReportando.infraccionesConfirmadas), 2)
miAcetatoInformativo.colorDeSemaforo(informacion['semaforo'][0])
# Draw the rectangles
for rectangulo in informacion['rectangulos']:
miAcetatoInformativo.colocarObjetivo(rectangulo[0], rectangulo[2])
if mostrarImagen:
#cv2.imshow('Visual', miAcetatoInformativo.aplicarAFrame(informacion['frame'])[120:239,60:360])
cv2.imshow(
'Visual',
miAcetatoInformativo.aplicarAFrame(informacion['frame']))
miAcetatoInformativo.inicializar()
tiempoEjecucion = time.time() - tiempoAuxiliar
#if tiempoEjecucion>periodoDeMuestreo:
miReporte.warning('Tiempo Afuera {0:2f}'.format(tiempoEjecucion) +
'[s] en frame {}'.format(frame_number))
#sys.stdout.write("\033[F")
while time.time() - tiempoAuxiliar < periodoDeMuestreo:
True
tiempoAuxiliar = time.time()
if (cambiosImportantes) | (numeroDeObjetos != len(
informacion['rectangulos'])):
miReporte.info(
'F{} Sema: '.format(frame_number) +
informacion['semaforo'][1] + ' I: ' +
str(miPoliciaReportando.numeroInfraccionesConfirmadas()) +
'/' + str(miPoliciaReportando.numeroInfraccionesTotales()) +
' Objetos: {}'.format(len(informacion['rectangulos'])))
numeroDeObjetos = len(informacion['rectangulos'])
porcentajeDeMemoria = psutil.virtual_memory()[2]
if porcentajeDeMemoria > 80:
miReporte.info('Estado de Memoria: ' + str(porcentajeDeMemoria) +
'/100')
if porcentajeDeMemoria > 92:
frameAOptimizar = min(informacionTotal)
miReporte.warning('Alcanzado 92/100 de memoria, borrando frame: ' +
str(frameAOptimizar))
del informacionTotal[frameAOptimizar]['recortados']
informacionTotal[frameAOptimizar]['recortados'] = {}
if porcentajeDeMemoria > 96:
miReporte.warning(
'Alcanzado 96/100 de memoria, borrando todo e inicializando')
del informacionTotal
informacionTotal = {}
frame_number = 0
frame_number += 1
if (frame_number >= topeEjecucion) & (topeEjecucion != 0):
miReporte.info(
'ABANDONANDO LA EJECUCION DE PROGRAMA por indice de auto acabado predeterminado'
)
break
if informacion['semaforo'][0] == -2:
miReporte.critical(
'ABANDONANDO LA EJECUCION DE PROGRAMA El semaforo ya no obtuvo señal, necesito recalibrar, abandonando la ejecución del programa'
)
break
ch = 0xFF & cv2.waitKey(5)
if ch == ord('q'):
miReporte.info(
'ABANDONANDO LA EJECUCION DE PROGRAMA por salida manual')
break
if ch == ord('s'):
cv2.imwrite(
datetime.datetime.now().strftime('%Y-%m-%d_%H:%M:%S') + '.jpg',
informacion['frame'])
fps.update()
# stop the timer and display FPS information
fps.stop()
def __main_function__():
# Import some global varialbes
global archivoDeVideo
global cambiosImportantes
cambiosImportantes = False
# Creamos el reporte inicial
miReporte = MiReporte(
levelLogging=logging.INFO, nombre=__name__
) # Se crea por defecto con nombre de la fecha y hora actual
miReporte.info(
'Programa iniciado exitosamente con ingreso de senal video ' +
archivoDeVideo + entradaReal + ' con semaforo ' +
semaforoSimuladoTexto + str(periodoDeSemaforo) + ', corriendo a ' +
str(mifps) + ' Frames por Segundo')
# Si no existe el directorio de reporte lo creo
if not os.path.exists(directorioDeReporte):
os.makedirs(directorioDeReporte)
# Is statements
if generarArchivosDebug:
miReporte.info('Generando Archivos de Debug')
else:
miReporte.info('Generando infracciones unicamente (No debug video)')
# If mostrar Imagenes
if mostrarImagen:
miReporte.info(
'Pantalla de muestra de funcionamiento en tiempo real encendida')
else:
miReporte.info(
'Pantalla de muestra de funcionamiento en tiempo real apagada')
# El directorio de reporte debe crearse al inicio del programa
# Variables de control:
numeroDeFrame = 0
maximoInfraccionesPorFrame = 20
#colores = np.random.randint(0,100,(maximoInfraccionesPorFrame,3))
# Cargando los parametros de instalacion:
# El archivo de video debe tener como minimo 5 caracteres para estar trnajando en modo simulado, de lo contrario estamos trabajando en modo real
if len(archivoDeVideo) > 4:
archivoParametrosACargar = archivoDeVideo[:-4] + '.npy'
else:
archivoParametrosACargar = 'datos.npy'
parametrosInstalacion = np.load(folderDeInstalacion + '/' +
archivoParametrosACargar)
miReporte.info('Datos de Instalacion de: ' + folderDeInstalacion + '/' +
archivoParametrosACargar)
poligonoSemaforo = parametrosInstalacion[0]
verticesPartida = parametrosInstalacion[1]
verticesLlegada = parametrosInstalacion[2]
angulo = parametrosInstalacion[3]
indicesSemaforo = SemaphoroParameters(poligonoSemaforo)
miReporte.info('Cargado exitosamente parametros de instalacion: ' +
str(parametrosInstalacion))
# Arrancando camara
if len(archivoDeVideo) == 0:
conVideoGrabado = False # modo real
if os.uname()[1] == 'alvarohurtado-305V4A':
miCamara = cv2.VideoCapture(1)
time.sleep(1)
elif os.uname()[1] == 'stanlee321-HP-240-G1-Notebook-PC':
miCamara = cv2.VideoCapture(0)
miCamara.set(cv2.CAP_PROP_FRAME_WIDTH, mediumRes[0])
miCamara.set(cv2.CAP_PROP_FRAME_HEIGHT, mediumRes[1])
time.sleep(1)
else:
miCamara = cv2.VideoCapture(1)
miCamara.set(cv2.CAP_PROP_FRAME_WIDTH, mediumRes[0])
miCamara.set(cv2.CAP_PROP_FRAME_HEIGHT, mediumRes[1])
#miCamara = VideoStream(src = 0, resolution = (1920,1080),poligono = poligonoSemaforo, debug = saltarFrames,fps = mifps, periodo = periodoDeSemaforo, gamma = gamma).start()
#miCamara = VideoStream(src = 0, resolution = (1280,960),poligono = poligonoSemaforo, debug = saltarFrames,fps = mifps, periodo = periodoDeSemaforo, gamma = gamma).start()
time.sleep(1)
miReporte.info('Activada Exitosamente cámara en tiempo real')
else:
conVideoGrabado = True
try:
miCamara = cv2.VideoCapture(directorioDeVideos + '/' +
archivoDeVideo)
miCamara.set(cv2.CAP_PROP_FRAME_WIDTH, mediumRes[0])
miCamara.set(cv2.CAP_PROP_FRAME_HEIGHT, mediumRes[1])
time.sleep(1)
miReporte.info('Archivo de video cargado exitosamente: ' +
directorioDeVideos + '/' + archivoDeVideo)
except Exception as currentException:
miReporte.error('No se pudo cargar el video por ' +
str(currentException))
# Load points for the semaphoro
# Se captura la imagen de flujo inicial y se trabaja con la misma
ret, frameVideo = miCamara.read()
frameFlujo = cv2.resize(frameVideo, (320, 240))
# Creación de objetos:
miPoliciaReportando = PoliciaInfractor(frameFlujo, verticesPartida,
verticesLlegada)
miGrabadora = GeneradorEvidencia(directorioDeReporte, mifps,
guardarRecortados)
miFiltro = IRSwitch()
miAcetatoInformativo = Acetato()
miSemaforo = CreateSemaforo(periodoDeSemaforo)
miAcetatoInformativo.colocarPoligono(np.array(poligonoSemaforo) // 2)
miAcetatoInformativo.colocarPoligono(np.array(verticesPartida))
miAcetatoInformativo.colocarPoligono(np.array(verticesLlegada))
fps = FPS().start()
informacionTotal = {}
frame_number = 0
tiempoAuxiliar = time.time()
periodoDeMuestreo = 1.0 / mifps
grupo = [0]
while True:
# LEEMOS LA CAMARA DE FLUJO
if conVideoGrabado:
for i in range(videofps // mifps):
ret, frameVideo = miCamara.read()
else:
ret, frameVideo = miCamara.read()
frameFlujo = cv2.resize(frameVideo, (320, 240),
interpolation=cv2.INTER_NEAREST)
tic = time.time()
informacionTotal[frame_number] = frameFlujo.copy()
pixeles = indicesSemaforo(frameVideo)
print('La longitud pixels: ', pixeles.shape)
senalSemaforo, semaforoLiteral, flanco, periodo = miSemaforo.obtenerColorEnSemaforo(
pixeles)
tac = time.time()
print('tic-tac', tac - tic)
if periodo != 0:
miReporte.info('SEMAFORO EN VERDE, EL PERIODO ES ' + str(periodo))
else:
pass
# Si tengo infracciones pendientes las evoluciono
if senalSemaforo >= 1: # Si estamos en rojo, realizamos una accion
if flanco == 1: # esto se inicia al principio de este estado
miReporte.info('SEMAFORO EN ROJO')
miPoliciaReportando.inicializarAgente()
del informacionTotal
informacionTotal = {}
frame_number = 0
else:
pass
cambiosImportantes, ondaFiltrada, flanco, flujoTotal = miPoliciaReportando.seguirImagen(
frame_number, frameFlujo)
else:
pass
if senalSemaforo == 0: # Si estamos en verde realizamos otra accion
if flanco == -1: # Si estamos en verde y en flanco, primer verde, realizamos algo
miReporte.info(
'Infracciones: ' +
str(miPoliciaReportando.numeroInfraccionesConfirmadas()))
if generarArchivosDebug:
miGrabadora.generarReporteInfraccion(
informacionTotal, False,
miPoliciaReportando.numeroInfraccionesConfirmadas())
if miPoliciaReportando.numeroInfraccionesConfirmadas() > 0:
infraccionEnRevision = miPoliciaReportando.popInfraccion()
miGrabadora.generarReporteInfraccion(informacionTotal,
infraccionEnRevision)
else:
#Si no hay infracciones a reportar me fijo el estado del filtro:
tiempoAhora = datetime.datetime.now(
).hour * 60 + datetime.datetime.now().minute
if (tiempoAhora > amaneciendo) & (miFiltro.ultimoEstado !=
'Filtro Activado'):
miFiltro.colocarFiltroIR()
if (tiempoAhora < amaneciendo) & (miFiltro.ultimoEstado !=
'Filtro Desactivado'):
miFiltro.quitarFiltroIR()
pass
# Draw frame number into image on top
for infraction in miPoliciaReportando.listaDeInfracciones:
for puntos in infraction['desplazamiento']:
puntosExtraidos = puntos.ravel().reshape(
puntos.ravel().shape[0] // 2, 2)
for punto in puntosExtraidos:
if infraction['estado'] == 'Confirmado':
miAcetatoInformativo.colocarPunto(tuple(punto), 0)
else:
miAcetatoInformativo.colocarPunto(tuple(punto), 1)
# Configs and displays for the MASK according to the semaforo
#miAcetatoInformativo.agregarTextoEn("I{}".format(miPoliciaReportando.infraccionesConfirmadas), 2)
miAcetatoInformativo.colorDeSemaforo(senalSemaforo)
if mostrarImagen:
#cv2.imshow('Visual', miAcetatoInformativo.aplicarAFrame(frameFlujo)[120:239,60:360])
cv2.imshow('Visual',
miAcetatoInformativo.aplicarAFrame(frameFlujo))
miAcetatoInformativo.inicializar()
tiempoEjecucion = time.time() - tiempoAuxiliar
if tiempoEjecucion > periodoDeMuestreo:
miReporte.warning('Tiempo Afuera {0:2f}'.format(tiempoEjecucion) +
'[s] en frame {}'.format(frame_number))
#sys.stdout.write("\033[F")
while time.time() - tiempoAuxiliar < periodoDeMuestreo:
True
tiempoAuxiliar = time.time()
if cambiosImportantes:
miReporte.info(
'F{} Sema: '.format(frame_number) + semaforoLiteral + ' I: ' +
str(miPoliciaReportando.numeroInfraccionesConfirmadas()) +
'/' + str(miPoliciaReportando.numeroInfraccionesTotales()))
porcentajeDeMemoria = psutil.virtual_memory()[2]
if (porcentajeDeMemoria > 80) & (os.uname()[1] == 'raspberrypi'):
miReporte.info('Estado de Memoria: ' + str(porcentajeDeMemoria) +
'/100')
if porcentajeDeMemoria > 92:
frameAOptimizar = min(informacionTotal)
miReporte.warning('Alcanzado 92/100 de memoria, borrando frame: ' +
str(frameAOptimizar))
del informacionTotal[frameAOptimizar]['recortados']
informacionTotal[frameAOptimizar]['recortados'] = {}
if porcentajeDeMemoria > 96:
miReporte.warning(
'Alcanzado 96/100 de memoria, borrando todo e inicializando')
del informacionTotal
informacionTotal = {}
frame_number = 0
frame_number += 1
if (frame_number >= topeEjecucion) & (topeEjecucion != 0):
miReporte.info(
'ABANDONANDO LA EJECUCION DE PROGRAMA por indice de auto acabado predeterminado'
)
break
if senalSemaforo == -2:
miReporte.critical(
'ABANDONANDO LA EJECUCION DE PROGRAMA El semaforo ya no obtuvo señal, necesito recalibrar, abandonando la ejecución del programa'
)
break
ch = 0xFF & cv2.waitKey(5)
if ch == ord('q'):
miReporte.info(
'ABANDONANDO LA EJECUCION DE PROGRAMA por salida manual')
break
if ch == ord('s'):
cv2.imwrite(
datetime.datetime.now().strftime('%Y-%m-%d_%H:%M:%S') + '.jpg',
frameFlujo)
fps.update()
# stop the timer and display FPS information
fps.stop()
