Top 7 pretrained-weight Open-Source Projects

pytorch-image-models

35 29,828 9.4 Python

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNet-V3/V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

Project mention: FLaNK AI Weekly 18 March 2024 | dev.to | 2024-03-18

segmentation_models.pytorch

14 8,844 4.1 Python

Segmentation models with pretrained backbones. PyTorch.

Project mention: Instance segmentation of small objects in grainy drone imagery | /r/computervision | 2023-12-09

Also, I’d suggest considering switching to the segmentation-models library - it provides U-Net models with a variety of pretrained backbones of as encoders. The author also put out a PyTorch version. https://github.com/qubvel/segmentation_models.pytorch https://github.com/qubvel/segmentation_models

InfluxDB

www.influxdata.com featured

Power Real-Time Data Analytics at Scale. Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.
super-gradients

8 4,332 9.5 Jupyter Notebook

Easily train or fine-tune SOTA computer vision models with one open source training library. The home of Yolo-NAS.

Project mention: Zero-Shot Prediction Plugin for FiftyOne | dev.to | 2024-03-13

Most computer vision models are trained to predict on a preset list of label classes. In object detection, for instance, many of the most popular models like YOLOv8 and YOLO-NAS are pretrained with the classes from the MS COCO dataset. If you download the weights checkpoints for these models and run prediction on your dataset, you will generate object detection bounding boxes for the 80 COCO classes.

Entity

2 668 6.9 Jupyter Notebook

EntitySeg Toolbox: Towards Open-World and High-Quality Image Segmentation
PyTorch_CIFAR10

1 599 0.0 Python

Pretrained TorchVision models on CIFAR10 dataset (with weights)
HIPT

1 455 4.3 Jupyter Notebook

Hierarchical Image Pyramid Transformer - CVPR 2022 (Oral)
SegmentationCpp

3 402 3.4 C++

A c++ trainable semantic segmentation library based on libtorch (pytorch c++). Backbone: VGG, ResNet, ResNext. Architecture: FPN, U-Net, PAN, LinkNet, PSPNet, DeepLab-V3, DeepLab-V3+ by now.
SaaSHub

www.saashub.com featured

SaaSHub - Software Alternatives and Reviews. SaaSHub helps you find the best software and product alternatives

NOTE: The open source projects on this list are ordered by number of github stars. The number of mentions indicates repo mentiontions in the last 12 Months or since we started tracking (Dec 2020).

pretrained-weights related posts

GitHub - Deci-AI/super-gradients: Easily train or fine-tune SOTA co...GitHub - Deci-AI/super-gradients: Easily train or fine-tune SOTA co...

1 project | /r/thirdbrain | 12 May 2023
Meet YOLO-NAS: An Open-Sourced YOLO-based Architecture Redefining State-of-the-Art in Object Detection

1 project | /r/machinelearningnews | 8 May 2023