NBStephens

datachain

DataChain 🔗 AI-dataframe to enrich, transform and analyze data from cloud storages for ML training and LLM apps

ssl-data-curation

PyTorch code for hierarchical k-means -- a data curation method for self-supervised learning

satmae_pp

Official repository for "Rethinking Transformers Pre-training for Multi-Spectral Satellite Imagery" (CVPR 2024)

unilm

Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities

clip-rocket

Code release for "Improved baselines for vision-language pre-training"

ijepa

Official codebase for I-JEPA, the Image-based Joint-Embedding Predictive Architecture. First outlined in the CVPR paper, "Self-supervised learning from images with a joint-embedding predictive architecture."

sam-hq

Segment Anything in High Quality [NeurIPS 2023]

what_are_embeddings

A deep dive into embeddings starting from fundamentals

Satellite-Image-Time-Series-Datasets

This page presents a list of satellite imagery datasets with a temporal dimension, mainly satellite image time series (SITS) and satellite videos, for various computer vision and deep learning tasks. It covers multi-temporal datasets with more than two acquisitions but not bi-temporal datasets.

SpectFormers

rembg

Rembg is a tool to remove images background

ultralytics

NEW - YOLOv8 🚀 in PyTorch > ONNX > OpenVINO > CoreML > TFLite

benchmark_VAE

Unifying Variational Autoencoder (VAE) implementations in Pytorch (NeurIPS 2022)

optformer

G.pt

Official PyTorch Implementation of "Learning to Learn with Generative Models of Neural Network Checkpoints"

lsnms

Large Scale Non maximum Suppression. This project implements a O(nlog(n)) approach for non max suppression, useful for object detection ran on very large images such as satellite or histology, when the number of instances to prune becomes very large.

over9000

Over9000 optimizer

techniques

Techniques for deep learning with satellite & aerial imagery

satellite-imagery-labeling-tool

This is a lightweight web-interface for creating and sharing vector annotations over satellite/aerial imagery scenes.

parallel_dbscan

Python module for the parallel dbscan based on NWU code

surgeon-pytorch

A library to inspect and extract intermediate layers of PyTorch models.

clipping-benchmarks

Benchmarking Sutherland-Hodgman clipping in a few languages

yoltv5

YOLT, now with PyTorch.

grip

Preview GitHub README.md files locally before committing them.

Detic

Code release for "Detecting Twenty-thousand Classes using Image-level Supervision".

umap

Uniform Manifold Approximation and Projection

DIT

[TNNLS] Full Transformer Framework for Robust Point Cloud Registration with Deep Information Interaction

Torch-Pruning

[CVPR 2023] Towards Any Structural Pruning; LLMs / SAM / Diffusion / Transformers / YOLOv8 / CNNs

SpectralCluster

Python re-implementation of the (constrained) spectral clustering algorithms used in Google's speaker diarization papers.

awesome-satellite-imagery-datasets

🛰️ List of satellite image training datasets with annotations for computer vision and deep learning