alexandercaldwell

VIGIA-PlumeNet

alexandercaldwell/VIGIA-PlumeNet

VIGIA-PlumeNet is a machine learning model from alexandercaldwell. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as cc0-1.0.

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.