Primary vein segmentation in grapevine leaves

Similar to a previous vein segmentaiton model (https://zenodo.org/records/17741713) that isolates all major veins, this model attempts to segment only the five major primary veins of a grapevine leaf (midvein, and the two distal and two proximal lateral veins) using a UNet model with the RGB segmented leaf mask and its associated Euler Characteristic Transform (ECT). All necessary data and outputs are included in this upload except for the folder of pre-segmented RGBA images of leaves segmented in the alpha channel called "FINAL_MASKS" that can be downloaded at https://zenodo.org/records/16883403 and placed into the main project folder here. Please note that 24,911 RGB leaf masks (segmented on a black background) and associated vein masks that are used for prediction found in ./data are not aligned to original FINAL_MASKS leaves, as derived in https://zenodo.org/records/17741713.

The following is a summary showing the relationship of this model to others:

Segmentation of RGB leaf masks

LEAF_SEGMENTATION: https://zenodo.org/records/16883916

FINAL_MASKS: https://zenodo.org/records/16883403

Segmentation of full vein mask

Vein segmentation of pre-segmented grapevine leaves using the radial Euler Characteristic Transform: https://zenodo.org/records/17741713

Segmentation of primary veins

Primary vein segmentation in grapevine leaves: https://zenodo.org/records/17881166

Segmentation of midvein

Midvein segmentation of grapevine leaves: https://zenodo.org/records/18005383

Gaussian heatmap estimation of petiolar junction coordinate

Grapevine leaf petiolar junction detection: https://zenodo.org/records/17922822

Segmentation of midvein, distal, and proximal lobes

Grapevine leaf lobe detection model: https://zenodo.org/records/18004938