我正在尝试训练一个 U-net 来对卫星数据进行图像分割,从而提取一个具有九种不同道路类型的道路网络。到目前为止,我已经尝试了许多在网络上免费提供的不同 U-net 代码,但是我无法根据我的具体情况定制它们。我真诚地希望你能帮助我。
卫星图像和相关标签可以通过以下链接下载: 卫星图像和相关标签
此外,我编写了以下代码来为 Unet 准备数据
import skimage
from skimage.io import imread, imshow, imread_collection, concatenate_images
from skimage.transform import resize
from skimage.morphology import label
import numpy as np
import matplotlib.pyplot as plt
from keras.models import Model
from keras.layers import Input, merge, Convolution2D, MaxPooling2D, UpSampling2D, Reshape, core, Dropout
from keras.optimizers import Adam
from keras.callbacks import ModelCheckpoint, LearningRateScheduler
from keras import backend as K
from sklearn.metrics import jaccard_similarity_score
from shapely.geometry import MultiPolygon, Polygon
import shapely.wkt
import shapely.affinity
from collections …Run Code Online (Sandbox Code Playgroud) python image-segmentation deep-learning conv-neural-network unity3d-unet