Esempio n. 1
0
args = vars(parser.parse_args())
count = args.get('count')
print 'We are gonna download %d negative images' % count

# We need the elements
print 'Loading %s...' % ELEMENTS_FILENAME
with open(ELEMENTS_FILENAME, 'r') as f:
    elements = json.load(f)

# Randomize elements list to make sure we don't download all pics from the
# same sample. Then cut it.
shuffle(elements)

# Now we're gonna download the satellite images for these locations
mapbox_static = MapboxStatic(
    namespace='negative',
    root_folder='satellite/negative')

total_downloaded = 0
for element in elements:
    if total_downloaded >= count:
            break
    if element.get('type') != 'node':
        continue
    # Move the latlon a random amount, random() is in the range [0.0, 1.0)
    target_lat = element.get('lat') + (random.random() - 0.5)
    target_lon = element.get('lon') + (random.random() - 0.5)
    url = mapbox_static.get_url(latitude=target_lat, longitude=target_lon)
    print url
    success = mapbox_static.download_tile(
        element_id=element.get('id'),
Esempio n. 2
0
# We need the elements
print 'Loading %s...' % ELEMENTS_FILENAME
with open(ELEMENTS_FILENAME, 'r') as f:
    elements = json.load(f)

# We need the elements
print 'Loading %s...' % WAYS_DATA_FILENAME
with open(WAYS_DATA_FILENAME, 'r') as f:
    ways_data = json.load(f)

# Randomize elements list to make sure we don't download all pics from the
# same sample
shuffle(elements)

# Now we're gonna download the satellite images for these locations
mapbox_static = MapboxStatic(namespace='pitch', root_folder='satellite')

total_downloaded = 0
for element in elements:
    # They're strings in the dict now
    element_id_str = unicode(element.get('id'))
    sport = element.get('tags', {}).get('sport', 'unknown').lower()
    if element_id_str in ways_data and sport == target_sport:
        if total_downloaded >= count:
            break
        print '> Element: %s (%s)' % (element.get('id'), sport)
        url = mapbox_static.get_url(
            latitude=ways_data[element_id_str].get('lat'),
            longitude=ways_data[element_id_str].get('lon'))
        print url
        element_id_sport = '%s_%s' % (sport, element_id_str)
Esempio n. 3
0
print 'Loading %s...' % ELEMENTS_FILENAME
with open(ELEMENTS_FILENAME, 'r') as f:
    elements = json.load(f)

# We need the elements
print 'Loading %s...' % WAYS_DATA_FILENAME
with open(WAYS_DATA_FILENAME, 'r') as f:
    ways_data = json.load(f)

# Randomize elements list to make sure we don't download all pics from the
# same sample
shuffle(elements)

# Now we're gonna download the satellite images for these locations
mapbox_static = MapboxStatic(
    namespace='pitch',
    root_folder='satellite')

total_downloaded = 0
for element in elements:
    # They're strings in the dict now
    element_id_str = unicode(element.get('id'))
    sport = element.get('tags', {}).get('sport', 'unknown').lower()
    if element_id_str in ways_data and sport == target_sport:
        if total_downloaded >= count:
            break
        print '> Element: %s (%s)' % (element.get('id'), sport)
        url = mapbox_static.get_url(
            latitude=ways_data[element_id_str].get('lat'),
            longitude=ways_data[element_id_str].get('lon'))
        print url
Esempio n. 4
0
def get_satellite(inputFile,
                  mapboxtoken=None,
                  count=1000,
                  zoomLevel=17,
                  outputFolder='data',
                  xpixel=480,
                  ypixel=360,
                  epsg=None,
                  elements=None,
                  randomImages=False):
    '''
	Get satellite data in order to input GIS information.

	Parameters:
	'inputFile': Input file (GeoJSON format or parsed into GeoJSON)
	'mapboxtoken': Access token for Mapbox (go to mapbox.com to create one)
	'count': Number of satellite images to be downloaded
	'zoomLevel': Zoom level (see libs/zoomLevel.csv for resolutions)
	'outputFolder': Folder to store output data in
	'xpixel': Number of pixels of satellite images (width)
	'ypixel': Number of pixels of satellite images (height)
	'epsg': EPSG code for coordinate system in GIS data (will try to find automatically 
		if not provided)
	'elements': GIS data can also be input directly
	'randomImages': Get center of random polygons (False) or within Boundary Box of data (True)
	'''
    if (not inputFile) and (not elements):
        print "Error: Provide input file."
        exit()
    if not mapboxtoken:
        print "Error: Provide mapbox token (more informations on www.mapbox.com)."
        exit()
    #parser.add_argument('--sport',
    #   type=str, default='baseball',
    #  help='Sport tag, for example: baseball, tennis, or soccer.')

    # We need the elements
    if not elements:
        print 'Loading %s...' % inputFile
        with open(inputFile, 'r') as f:
            elements = json.load(f)

    #get coordinate system
    myCoordConvert = CoordConvert()
    code = myCoordConvert.getCoordSystem(elements, epsg)
    #create folders
    subpath = outputFolder + "/" + os.path.split(inputFile)[-1][:-5]
    if not os.path.isdir(subpath):
        os.mkdir(subpath)
        print 'Directory', subpath, 'created'
    if not os.path.isdir(subpath + satDataFolder):
        os.mkdir(subpath + satDataFolder)
        print 'Directory', subpath + satDataFolder, 'created'
    if not os.path.isdir(subpath + testDataFolder):
        os.mkdir(subpath + testDataFolder)
        print 'Directory', subpath + testDataFolder, 'created'

    #Write metadata
    with open(subpath + satDataFolder + "meta.csv", "a+") as f:
        f.write("ZoomLevel,," + str(zoomLevel) + "\n")
    #get bbox if set to random
    if randomImages:
        xlist = []
        ylist = []
        for element in elements['features']:
            minxe, maxxe, minye, maxye = getBBox(element)
            xlist.append(minxe)
            xlist.append(maxxe)
            ylist.append(minye)
            ylist.append(maxye)
        minx = min(xlist)
        maxx = max(xlist)
        miny = min(ylist)
        maxy = max(ylist)

    element_list = []
    index_list = range(len(elements['features']))  #featue map
    # Randomize elements list to make sure we don't download all pics from the
    shuffle(index_list)
    for i in index_list:
        element_list.append(elements['features'][i])  #feature map

    # Now we're gonna download the satellite images for these locations
    namespace = os.path.split(
        inputFile)[-1][:-5]  #get input file name as namespace

    mapbox_static = MapboxStatic(namespace=namespace,
                                 root_folder=subpath + satDataFolder[0:-1])

    total_downloaded = 0
    c = 0
    print "------------------- Getting Satellite data -------------------"
    for element in element_list:
        if randomImages:
            randomValue = random()
            av_lon = minx + ((maxx - minx) * randomValue)
            av_lat = miny + ((maxy - miny) * randomValue)
            element_id_str = 1000000 + c  #1000000 indicates random value
            with open(subpath + satDataFolder + "meta.csv", "a+") as f:
                f.write(
                    str(element_id_str) + "," + str(av_lon) + "," +
                    str(av_lat) + "\n")
        else:
            element_id_str = index_list[c]
            #figure out center of polygon
            av_lon, av_lat = latLon(element)
        #Convert to standard format
        if code != 4319:  # if not already in wgs84 standard format
            lotlan = myCoordConvert.convert(av_lon, av_lat)
            longitude = lotlan[0]
            latitude = lotlan[1]
        else:  #if already in wgs84 format
            latitude = av_lat
            longitude = av_lon

        #get url
        print "Coordinates WSG64: " + str(longitude) + ',' + str(latitude)
        if (av_lon != longitude) and (av_lat != latitude):
            print "Coordinates Native: " + str(av_lon) + ',' + str(av_lat)
        url = mapbox_static.get_url(latitude=latitude,
                                    longitude=longitude,
                                    mapbox_zoom=zoomLevel,
                                    access_token=mapboxtoken,
                                    width=xpixel,
                                    height=ypixel)
        #download data
        success = mapbox_static.download_tile(element_id=element_id_str,
                                              url=url,
                                              verbose=True)
        if success:
            total_downloaded += 1
            print total_downloaded, '/', count
        c += 1
        if total_downloaded >= count:
            break