Writing Data

Objects read in by deepforest.utilities.read_file are geopandas GeoDataFrames. They can be exported as CSV files or shapefiles.

annotations.to_csv("annotations.csv")
annotations.to_file("annotations.shp")

Converting from Image to Geographic Coordinates

Standard computer vision models have no concept of geographic information, which is why we use the image coordinate system to represent coordinates within DeepForest. If you want to convert predictions back into geographic coordinates, we provide a utility to go from image to geo coordinates based on the coordinate reference system of the file in the image_path column.

import geopandas as gpd
import rasterio as rio
from matplotlib import pyplot as plt
from shapely import geometry

from deepforest import get_data
from deepforest import utilities

annotations = get_data("2018_SJER_3_252000_4107000_image_477.csv")
path_to_raster = get_data("2018_SJER_3_252000_4107000_image_477.tif")
src = rio.open(path_to_raster)
original = utilities.read_file(annotations)

geo_coords = utilities.image_to_geo_coordinates(original)
src_window = geometry.box(*src.bounds)

fig, ax = plt.subplots(figsize=(10, 10))
gpd.GeoSeries(src_window).plot(ax=ax, color="blue", alpha=0.5)
geo_coords.plot(ax=ax, color="red")
plt.show()