nasaharvest / presto

Lightweight, Pre-trained Transformers for Remote Sensing Timeseries
https://arxiv.org/abs/2304.14065
MIT License
151 stars 26 forks source link

Update TreeSat to be multilabel #5

Closed gabrieltseng closed 1 year ago

gabrieltseng commented 1 year ago

As pointed out in #3, although TreeSat is multi-label dataset, we were only taking the first label (in the filename) instead of all the labels in the labels json.

This PR updates the models to be multilabel, with the following results (paper to be updated):

Screenshot 2023-05-03 at 8 05 37 AM