File size: 1,749 Bytes
952641a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
import os
import cv2
import zipfile
import shapefile
import numpy as np
from io import BytesIO
from shapely.geometry import Polygon
import matplotlib.pyplot as plt
from PIL import Image
def convert_pixel_to_latlon(x, y, image_width, image_height, image_coords):
top_left, top_right, bottom_right, bottom_left = image_coords
lon_top = top_left[0] + (top_right[0] - top_left[0]) * (x / image_width)
lon_bottom = bottom_left[0] + (bottom_right[0] - bottom_left[0]) * (x / image_width)
lat_left = top_left[1] + (bottom_left[1] - top_left[1]) * (y / image_height)
lat_right = top_right[1] + (bottom_right[1] - top_right[1]) * (y / image_height)
lon = lon_top + (lon_bottom - lon_top) * (y / image_height)
lat = lat_left + (lat_right - lat_left) * (x / image_width)
return lon, lat
# Function to create a shapefile with image dimensions and bounding boxes
def create_shapefile_with_latlon(bboxes, image_shape, image_coords, shapefile_path):
w = shapefile.Writer(shapefile_path)
w.field('id', 'C')
img_width, img_height = image_shape
# Add bounding boxes for weeds
for idx, (x1, y1, x2, y2) in enumerate(bboxes):
top_left = convert_pixel_to_latlon(x1, y1, img_width, img_height, image_coords)
top_right = convert_pixel_to_latlon(x2, y1, img_width, img_height, image_coords)
bottom_left = convert_pixel_to_latlon(x1, y2, img_width, img_height, image_coords)
bottom_right = convert_pixel_to_latlon(x2, y2, img_width, img_height, image_coords)
poly = Polygon([top_left, top_right, bottom_right, bottom_left, top_left])
w.poly([poly.exterior.coords])
w.record(f'weed_{idx}')
w.close() |