Skip to content
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

U-Net with customized c2smsfloods benchmark dataset for linear probing and evaluation #87

Open
wants to merge 28 commits into
base: main
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
28 commits
Select commit Hold shift + click to select a range
98cbe84
WIP notebook for unet w customized c2smsfloods benchmark dataset
lillythomas Dec 12, 2023
07b183a
working dataloader and minimal unet
lillythomas Dec 14, 2023
8e23453
tiny experiment trained on 3 epochs on subset of data, plus predictio…
lillythomas Dec 14, 2023
c5335cd
normalize pixels
lillythomas Dec 14, 2023
10c27f9
plot some predictions after tiny 3 epoch training experiment with 100…
lillythomas Dec 14, 2023
a90134e
datamodule and model for finetuning
lillythomas Dec 20, 2023
45876e1
remove extra iteration
lillythomas Dec 20, 2023
c42d13b
working datamodule for finetuning, embeddings integrated in unet
lillythomas Dec 21, 2023
4a59071
remove softmax
lillythomas Dec 21, 2023
11b1952
remove comma, change loss function, update decoder, update clay cpkt …
lillythomas Dec 21, 2023
8847807
update notebook with train from scratch and label plot
lillythomas Dec 22, 2023
0e0bbb3
update notebook with train from scratch and label plot
lillythomas Dec 22, 2023
02fd4d7
remove tensor grad true assignment which fixes the loss convergence i…
lillythomas Jan 2, 2024
df16b97
add cloud and cloud shadow masking, loss function updates
lillythomas Jan 11, 2024
888182c
loss fn updates
lillythomas Jan 11, 2024
acfedf4
import eval dm
lillythomas Jan 11, 2024
bdf09cc
adjust to use bceloss
lillythomas Jan 11, 2024
25ad052
add assert statements to ensure grad enabled
lillythomas Jan 11, 2024
4d867b3
turn off grads for clay, fix BCEloss, add IoU eval
lillythomas Jan 12, 2024
ad3a694
reduce nb size by clearing most outputs
lillythomas Jan 12, 2024
8bcc685
working models and tensorboard logging
lillythomas Jan 13, 2024
218c540
working models and tensorboard logging
lillythomas Jan 13, 2024
7292436
rename notebook, add early stopping for clay finetuned variant
lillythomas Jan 13, 2024
f37284b
rename notebook, add early stopping for clay finetuned variant
lillythomas Jan 13, 2024
241374a
add clay decoder
lillythomas Jan 15, 2024
5310e3e
add shuffling with seed
lillythomas Jan 15, 2024
3aa688c
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Jan 15, 2024
c990ee5
Merge branch 'main' into benchmark_segmentation_task
brunosan Feb 15, 2024
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
1,563 changes: 1,563 additions & 0 deletions notebooks/c2smsfloods_unet.ipynb

Large diffs are not rendered by default.