Skip to content
#

pretrained-weights

Star

Here are 49 public repositories matching this topic...

Language: All
Filter by language
All 49 Python 29 Jupyter Notebook 15 C++ 2
Sort: Best match
Sort options
Best match Most stars Fewest stars Most forks Fewest forks Recently updated Least recently updated

rwightman / pytorch-image-models

Sponsor
Star 18.1k

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more

pytorch resnet pretrained-models mixnet pretrained-weights imagenet-classifier distributed-training dual-path-networks cnn-classification mobilenet-v2 mnasnet mobile-deep-learning mobilenetv3 efficientnet augmix randaugment efficientnet-training nfnets normalization-free-training vision-transformer-models
  • Updated Apr 28, 2022
  • Python

Deci-AI / super-gradients

Star 329
Open

How to train object detection model or classification model on custom dataset?

1
karndeepsingh
karndeepsingh commented Jan 20, 2022

Hi,
I have image data annotated for object detection as well on classification task. How I can train and build a object detection model and classification model on my own dataset?

A formal guideline would help me to understand the process thanks. Basically, I want to know how we can load our custom data into the Dataloader in supergradient?

Read more
good first issue

AllentDan / LibtorchSegmentation

Star 237

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.

neural-network cpp models pytorch imagenet resnet image-segmentation unet semantic-segmentation resnext pretrained-weights pspnet fpn deeplabv3 deeplabv3plus libtorch pytorch-cpp pytorch-cpp-frontend pretrained-backbones libtorch-segment
  • Updated Nov 17, 2021
  • C++

yang-ruixin / pytorch-image-models-with-simclr

Star 13

Multi-label classification based on timm, and add SimCLR to timm.

pytorch semi-supervised-learning transfer-learning pretrained-models multi-label-classification multi-task mixnet multi-label pretrained-weights multi-task-learning semi-supervised imagenet-classifier self-supervised self-supervised-learning mobilenetv3 efficientnet simclr gradient-centralization vision-transformer-models timm
  • Updated Sep 13, 2021
  • Jupyter Notebook

Improve this page

Add a description, image, and links to the pretrained-weights topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the pretrained-weights topic, visit your repo's landing page and select "manage topics."

Learn more