-
Notifications
You must be signed in to change notification settings - Fork 321
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
GASSL ResNet50 Weights #1325
GASSL ResNet50 Weights #1325
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Need to add to docs/api/resnet_pretrained_weights.csv
. We should start thinking about how we want to handle multiple satellites. I'm thinking we'll want different tables for each satellite since the benchmark datasets will differ. We can punt that to the PR where we add all our new SSL4EO-L models.
|
acb8be3
to
fe3948d
Compare
No idea why coverage is failing |
This LGTM, @adamjstewart any problems with merging this? |
Give me some time to review. How urgent is this? Would prefer to focus on the paper until the deadline, but can review sooner if needed. I mostly want to verify the transforms and filename and such. |
Do we want to call it fMoW or Maxar? Might be more clear what the weights can be used for with the latter. |
The current name fits with the existing scheme |
The existing scheme uses image source, not dataset. |
Fmow RGB is more than just Maxar though. |
Why did you choose Apache 2.0 for the license? |
Also, should we confirm the license of the pre-trained model before we modify and redistribute it? If it's unlicensed then technically we can't use it. If @calebrob6 doesn't care then I don't care, it's Microsoft that will take the blame. |
Where are you seeing "Apache 2.0"? |
It seems the weights aren't licensed -- but just to be clear, we are not uploading the weights to this repo. It is always good to ask the authors though. |
It seems the weights are licensed here (https://zenodo.org/record/7379715/) as Creative Commons Attribution 4.0 International, which is quite open. |
I'll change the license on HF |
Already did |
This reverts commit bc33326.
This PR adds the ResNet50 RGB weights trained on fMoW (mostly Maxar WorldView imagery) using the Geography Aware Self-supervised Learning (GASSL) method. This is basically a modified MoCo-v2 with Geolocation as a pretext task (Geo) as well as temporal pairs contrastive learning (TP).
Weights taken from this repo and uploaded to huggingface.
Adding these weights is essential since it's being compared as a baseline in both the SatMAE and Scale-MAE papers.