#

pretrained-weights

Here are 52 public repositories matching this topic...

rwightman / pytorch-image-models

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 Jun 13, 2022
Python

qubvel / segmentation_models.pytorch

Segmentation models with pretrained backbones. PyTorch.

models image-processing pytorch neural-networks imagenet segmentation pretrained-models image-segmentation unet semantic-segmentation pretrained-weights pspnet fpn deeplabv3 unet-pytorch deeplab-v3-plus linknet segmentation-models unetplusplus pretrained-backbones

Updated Jun 14, 2022
Python

qubvel / classification_models

Classification models trained on ImageNet. Keras.

keras vgg imagenet densenet resnet pretrained-models inceptionv3 resnext pretrained-weights imagenet-classifier mobilenet classification-model senet xception nasnet inception-resnet-v2 squeeze-and-excitation

Updated Apr 7, 2020
Python

alvinwan / neural-backed-decision-trees

Making decision trees competitive with neural networks on CIFAR10, CIFAR100, TinyImagenet200, Imagenet

pytorch neural-networks imagenet image-classification pretrained-models decision-trees cifar10 interpretability pretrained-weights cifar100 tiny-imagenet explainability neural-backed-decision-trees

Updated Jun 3, 2021
Python

pythonlessons / TensorFlow-2.x-YOLOv3

YOLOv3 implementation in TensorFlow 2.3.1

tracking detection yolo darknet tensorrt pretrained-weights yolov3 tensorflow2 yolov4 yolov4-training

Updated May 26, 2022
Jupyter Notebook

CVLAB-Unibo / Real-time-self-adaptive-deep-stereo

Code for "Real-time self-adaptive deep stereo" - CVPR 2019 (ORAL)

tensorflow stereo-vision pretrained-weights unsupervised-machine-learning domain-adaptation dispnet online-adaptation madnet cvpr2019 cvpr2019-oral deep-stereo-network

Updated May 26, 2022
Python

huyvnphan / PyTorch_CIFAR10

Pretrained TorchVision models on CIFAR10 dataset (with weights)

computer-vision pytorch deeplearning pretrained-models cifar10 pretrained-weights torchvision

Updated Jan 19, 2022
Python

Deci-AI / super-gradients

Open

The mAP reported by SG is not consistent with COCO-API

1

NatanBagrov commented Mar 14, 2022

Describe the bug

When training a detection model, e.g. SSD Lite MobileNet, the mAP shown in SG (TensorBoard) is much higher than the mAP returned via COCO-API

To Reproduce

Steps to reproduce the behavior:

Branch https://github.com/Deci-AI/super-gradients/tree/ssd_mobilenet
Run train_from_recipe.py with coco_ssd_lite_mobilenet_v2.yaml or coco_ssd_mobilenet_v1.yaml
(

Read more

good first issue

Open

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

1

dvlab-research / Entity

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

computer-vision deep-learning cnn pytorch segmentation object-detection pretrained-models image-segmentation semantic-segmentation pretrained-weights instance-segmentation panoptic-segmentation fcos pretraining detectron2 condinst

Updated Apr 25, 2022
Python

AllentDan / LibtorchSegmentation

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.

Updated May 19, 2022
C++

jfhealthcare / Chexpert

Top 1 solution of Chexpert

pretrained-weights pytorch-implementation chexpert

Updated May 26, 2022
Python

donydchen / ganimation_replicate

An Out-of-the-Box Replication of GANimation using PyTorch, pretrained weights are available!

pytorch gan pretrained-models pretrained-weights out-of-the-box replicate action-units pretrained-checkpoints ganimation ganimation-pytorch

Updated May 25, 2021
Python

ndrplz / dilation-tensorflow

A native Tensorflow implementation of semantic segmentation according to Multi-Scale Context Aggregation by Dilated Convolutions (2016). Optionally uses the pretrained weights by the authors.

deep-learning tensorflow segmentation semantic-segmentation pretrained-weights dilation-tensorflow

Updated Jun 20, 2018
Python

GalacticExchange / pretrained

Pretrained is the most complete and frequently updated list of pretrained top-performing models. Tensorflow, Theano and others. Want to add your model? File an issue, and we will add it.

deep-learning pretrained-models pretrained-weights

Updated Dec 4, 2017

chuanenlin / drone-net

https://towardsdatascience.com/tutorial-build-an-object-detection-system-using-yolo-9a930513643a

dataset drones object-detection darknet pretrained-weights yolov3

Updated Oct 1, 2020
Python

Richarizardd / Self-Supervised-ViT-Path

Self-Supervised Vision Transformers Learn Visual Concepts in Histopathology (LMRL Workshop, NeurIPS 2021)

deep-learning pytorch transfer-learning unsupervised-learning weakly-supervised-learning pretrained-weights computational-pathology self-supervised-learning histopathology neurips vision-transformer

Updated Jun 9, 2022
Jupyter Notebook

mahmoodlab / HIPT

Hierarchical Image Pyramid Transformer - CVPR 2022 (Oral)

deep-learning pytorch transfer-learning high-resolution unsupervised-learning hierarchical-attention-networks cvpr weakly-supervised-learning pretrained-weights computational-pathology self-supervised-learning histopathology vision-transformer cvpr2022

Updated Jun 13, 2022
Jupyter Notebook

yang-ruixin / PyTorch-Image-Models-Multi-Label-Classification

Multi-label classification based on timm.

pytorch densenet resnet transfer-learning pretrained-models multi-label-classification multi-task mixnet multi-label pretrained-weights multi-task-learning imagenet-classifier mnasnet mobilenetv3 efficientnet hrnet regnet gradient-centralization vision-transformer-models timm

Updated Sep 13, 2021
Jupyter Notebook

AllentDan / LibtorchDetection

C++ trainable detection library based on libtorch (or pytorch c++). Yolov4 tiny provided now.

cpp detection pytorch yolo pretrained-weights pretrained detection-model libtorch pytorch-cpp yolov4 yolov4-tiny

Updated May 19, 2022
C++

OanaIgnat / I3D_Keras

I3D implemetation in Keras + video preprocessing + visualization of results

flow computer-vision deep-learning keras preprocessing action-recognition video-data pretrained-weights i3d i3d-inception-architecture kinetics-datasets

Updated May 26, 2022
Python

ASR

vietai / ASR

End-to-End Vietnamese Speech Recognition using wav2vec 2.0

asr pretrained-weights ctc-loss asr-model end-to-end-speech-recognition wav2vec2

Updated Sep 3, 2021

ShowLo / MobileNetV3

An implementation of MobileNetV3 with pyTorch

Updated Jul 24, 2020
Python

avoroshilov / tf-squeezenet

TensorFlow version of SqueezeNet with converted pretrained weights

neural-network tensorflow squeezenet pretrained-weights tf-squeezenet adversarial-example

Updated Mar 11, 2017
Python

linksense / EfficientNet.PyTorch

Concise, Modular, Human-friendly PyTorch implementation of EfficientNet with Pre-trained Weights.

pytorch imagenet pretrained-weights efficient-inference efficientnet efficient-model efficientnet-pytorch efficientnet-pretrained efficientseg

Updated Feb 22, 2020
Python

hoangthang1607 / nfnets-Tensorflow-2

Pre-trained NFNets with 99% of the accuracy of the official paper "High-Performance Large-Scale Image Recognition Without Normalization".

keras image-classification deepmind pretrained-weights tensorflow2 weight-standardization nfnets sgd-adc

Updated Feb 26, 2021
Jupyter Notebook

shubham-shahh / Open-Source-Models

Address book for computer vision models.

open-source opencv deep-neural-networks tensorflow keras pytorch detector object-detection transfer-learning pretrained-models pretrained-weights yolov3 tensorflow2 yolov4 yolov4-tiny

Updated May 9, 2022

CellEight / Pytorch-Adaptive-Instance-Normalization

A Pytorch implementation of the 2017 Huang et. al. paper "Arbitrary Style Transfer in Real-time with Adaptive Instance Normalization"

machine-learning computer-vision deep-learning paper pytorch style-transfer neural-networks datasets deep-learning-papers pretrained-weights huang

Updated Jul 13, 2020
Python

Lornatang / YOLOv3-PyTorch

Pytorch implements yolov3.Good performance, easy to use, fast speed.

simple detection inference easy-to-use image-augmentation pretrained-weights custom-dataset yolov3 yolov3-pytorch darknet-conversion

Updated May 1, 2020
Python

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

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

syt2 / CNN

pytorch implementation of several CNNs for image classification

cnn pytorch imagenet image-classification resnet cifar resnext pretrained-weights senet cbam sknet train-from-scratch

Updated Jul 12, 2020
Python

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."