EIDU-ROJ » 2017 » Optimización de algoritmo para segmentación de señales satelitales

#!/usr/bin/python

import sys, os, datetime

import os

import numpy as np

import h5py        #

#matplotlib.use('Agg')

from matplotlib.pylab import *

import matplotlib.pyplot as plt

from mpl_toolkits.basemap import Basemap

from scipy import *

from scipy import fftpack

from scipy import ndimage

from scipy import optimize

from optparse import OptionParser

from gnuradio.eng_option import eng_option

##import beacon_conf as c #Archivo de Configuracion

import time

a=np.zeros((10,10))

for i in range(3,len(a)):

    for j in range(0,len(a)-3):

        if j==i-3:

            a[i][j]=1

        if j==i-2:

            a[i][j]=1

        if j==i:

            a[i][j]=1

        if j==i+1:

            a[i][j]=1

a[4][5]=0

lbl=ndimage.label(a)[0]

slice_x, slice_y = ndimage.find_objects(lbl==2)[0]

roi = lbl[slice_x, slice_y]

plt.imshow(roi)

print lbl

print roi