- Image segmentation
- A tensorflow_2.0 implementation of U-Net.
- Python >= 3.6
- TensorFlow == 2.1.0
- opencv-python == 4.1.0.25
- dataset_cell数据集下载地址:https://pan.baidu.com/s/1PfOMxGUwJhO-9p8yDsasag 提取码:nqhh
- dataset_sidewalk数据集下载地址:https://pan.baidu.com/s/1SZ42yGLNf2C9XUNUQkQCbw 提取码:ntyh
- dataset_trafficRoad数据集下载地址:https://pan.baidu.com/s/1mFFAx7dPcrdkXsbGQomGRg 提取码:20oh
- 代码中 model = tf.keras.models.load_model('models/model-cell-0017-0.9356.h5')改成自己训练出来的模型名称。
- 代码中 model = tf.keras.models.load_model('models/model-sidewalk-0016-0.9945.h5')改成自己训练出来的模型名称。
- 代码中 model = tf.keras.models.load_model('models/model-trafficRoad-0017-0.9052.h5')改成自己训练出来的模型名称。