isaaccorley

isaaccorley

isaaccorley

isaaccorley

# pip install torch lightning scikit-learn numpy tqdm faissknn
import lightning.pytorch as pl
import numpy as np
import torch
from faissknn import FaissKNNClassifier
from lightning.pytorch.utilities import rank_zero_only
from sklearn.metrics import accuracy_score, f1_score, precision_score, recall_score
from tqdm import tqdm

isaaccorley

# download test set
mkdir -p fair1m/train/part1
mkdir fair1m/train/part2
mkdir fair1m/validation
mkdir fair1m/test

cd fair1m/train/part1
gdown 1LWT_ybL-s88Lzg9A9wHpj0h2rJHrqrVf # train/part1/images.zip
gdown 1CnOuS8oX6T9JMqQnfFsbmf7U38G6Vc8u # train/part1/labelXml.zip
cd ..

isaaccorley

isaaccorley

import torch
import torch.nn as nn
import torch.optim as optim
import segmentation_models_pytorch as smp
from torchmetrics import Accuracy
from tqdm import tqdm

from torchgeo.datasets import ETCI2021
from torchgeo.datamodules import ETCI2021DataModule

isaaccorley

isaaccorley

wsl --shutdown
diskpart
select vdisk file="C:\Users\<user>\AppData\Local\Packages\CanonicalGroupLimited.Ubuntu20.04onWindows_79rhkp1fndgsc\LocalState\ext4.vhdx"
attach vdisk readonly
compact vdisk
detach vdisk
exit