Skip to content

This is an example on how to train and evaluate a basic Unet model

License

Notifications You must be signed in to change notification settings

williamlidberg/Unet-tutorial

Repository files navigation

License: MIT Twitter Follow Python 3.7

U-net-tutorial

This is an example of how to implement transfer learning for semantic segmentation. Lunar impact craters are used to pre-train a UNet model, which is then trained on high-resolution LiDAR data and manually digitised hunting pits.

hunting pit
Hunting pit in a high resolution digital elevation model (DEM).

Click this button to get started

Open In Colab

Data description

Impact craters

Impact craters from the moon were used to pre-train the model. These creaters were digitised by NASA and are avalible from the Moon Crater Database v1 Robbins:https://astrogeology.usgs.gov/search/map/Moon/Research/Craters/lunar_crater_database_robbins_2018 The database contains approximately 1.3 million lunar impact craters and is approximately complete for all craters larger than about 1–2 km in diameter. Craters were manually identified and measured on Lunar Reconnaissance Orbiter Camera Wide-Angle Camera (WAC) images, in LRO Lunar Orbiter Laser Altimeter topography, SELENE Kaguya Terrain Camera images, and a merged LOLA+TC DTM.

The Moon LRO LOLA DEM 118m v1 was used as digital elevation model. This digital elevation model is based on data from the Lunar Orbiter Laser Altimeter, an instrument on the National Aeronautics and Space Agency Lunar Reconnaissance Orbiter spacecraft. The created DEM represents more than 6.5 billion measurements gathered between July 2009 and July 2013, adjusted for consistency in the coordinate system described below, and then converted to lunar radii. Source: https://astrogeology.usgs.gov/search/details/Moon/LRO/LOLA/Lunar_LRO_LOLA_Global_LDEM_118m_Mar2014/cub

impact crater
Impact crater in the lunar DEM

Hunting pits

Coordinates of known hunting pits were downloaded from the Swedish national heritage board. The coordinates were manually adjusted based on visual observations in a hill-shaded elevation model. The area surrounding known hunting pits was inspected and additional hunting pits were manually digitized if discovered. The adjusted coordinates were also converted to polygon circles outlining the size of each hunting pit. In total, 2519 hunting pits were digitized in this way. These polygons were converted into segmentation masks where pixels inside the polygons were given a value of 1 and pixels outside the polygons were given a value of 0.

hunting pits
In total, 2519 hunting pits were manually mapped in northern Sweden. Some 80% of the hunting pits were used for training the models while 20% were used for testing.

Links

PyPI: https://pypi.org/project/whitebox/

Contact: [email protected]

About

This is an example on how to train and evaluate a basic Unet model

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published