alexandercaldwell/VIGIA-PlumeNet

# VIGIAPlumeNet

This repository contains a PyTorch model for image segmentation of volcanic plumes, based on DinoV2, trained on the VIGIA Plume-data dataset. 
The model predicts the class of an input image from 6 categories. 

For a fast usage, take the traced version (not required to load DinoV2, etc.)
For training or for a more hands-on usage, use the regular version.

Check the full repo on gitlab https://gricad-gitlab.univ-grenoble-alpes.fr/vigia/vigia-plumenet