DavidLandup0 / deepvision
PyTorch and TensorFlow/Keras image models with automatic weight conversions and equal API/implementations - Vision Transformer (ViT), ResNetV2, EfficientNetV2, NeRF, SegFormer, MixTransformer, (planned...) DeepLabV3+, ConvNeXtV2, YOLO, etc.
β35Updated last year
Related projects β
Alternatives and complementary repositories for deepvision
- A Keras implementation of hybrid efficientnet swin transformer model.β33Updated last year
- This repository demonstrates how to use TensorFlow based SegFormer model in π€ transformers package.β30Updated 2 years ago
- Implementation of Deep Orthogonal Fusion of Local and Global Features in TensorFlow 2β25Updated last year
- ResNet-RS models rewritten in Tensorflow / Keras functional API.β27Updated last year
- Includes PyTorch -> Keras model porting code for DeiT models with fine-tuning and inference notebooks.β41Updated 2 years ago
- Tensorflow 2.0 Implementation of GCViT: Global Context Vision Transformerβ27Updated 10 months ago
- β24Updated 2 years ago
- A Detection Toolbox for Tensorflow2β56Updated last year
- Heatmap Learner Convolutional Neural Network for Object Counting and Localizationβ42Updated 9 months ago
- TensorFlow2.x implementation of RetinaNetβ40Updated 3 years ago
- Adds timm pretrained backbone to pytorch's FasterRcnn modelβ12Updated 9 months ago
- β16Updated last year
- Code for our paper CPPE - 5 (Medical Personal Protective Equipment), a new challenging object detection datasetβ67Updated 2 months ago
- Unofficial implementation of "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows" (https://arxiv.org/abs/2103.14030)β65Updated 3 years ago
- Includes PyTorch -> Keras model porting code for ConvNeXt family of models with fine-tuning and inference notebooks.β100Updated 2 years ago
- Converting Instance segmentation labels in COCO format to YOLOv5-segβ13Updated last year
- Implementation of RegNetY in TensorFlow 2β20Updated last year
- Implementation of Swin Transformers in TensorFlow along with converted pre-trained models, code for off-the-shelf classification and fineβ¦β56Updated 2 years ago
- Building a maskrcnn from scratch using tensorflow and kerasβ50Updated 3 years ago
- This repository shows how to train a custom detection model with the TFOD API, optimize it with TFLite, and perform inference with the opβ¦β31Updated 3 years ago
- Implements RNNPool and SoftPool for CNNs.β14Updated 3 years ago
- Keras Implementation of Vision Transformer (An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale)β95Updated 4 years ago
- A Tensorflow implementation of Deep Lab V3 Plus from scratch.β22Updated 4 years ago
- Python scripts performing object detection using the YOLOv6 model in ONNX.β93Updated 2 years ago
- Tensorflow implementation of the Vision Transformer (An Image is Worth 16x16 Words: Transformerβ97Updated 3 years ago
- A personal implementation of YOLOv5 (v6.0)β47Updated last year
- A clean, modular implementation of the Yolov7 model family, which uses the official pretrained weights, with utilities for training the mβ¦β116Updated 9 months ago
- Vision Transformers for image classification, image segmentation, and object detection.β43Updated last month
- Implementation of CaiT models in TensorFlow and ImageNet-1k checkpoints. Includes code for inference and fine-tuning.β12Updated last year
- PyTorch Lightning based training of Semantic Segmentation modelsβ38Updated last year