mmdetection VS pytorch-image-models

Compare mmdetection vs pytorch-image-models and see what are their differences.

Our great sponsors
  • InfluxDB - Power Real-Time Data Analytics at Scale
  • WorkOS - The modern identity platform for B2B SaaS
  • SaaSHub - Software Alternatives and Reviews
mmdetection pytorch-image-models
23 35
27,742 29,751
2.3% 2.9%
8.7 9.4
8 days ago 3 days ago
Python Python
Apache License 2.0 Apache License 2.0
The number of mentions indicates the total number of mentions that we've tracked plus the number of user suggested alternatives.
Stars - the number of stars that a project has on GitHub. Growth - month over month growth in stars.
Activity is a relative number indicating how actively a project is being developed. Recent commits have higher weight than older ones.
For example, an activity of 9.0 indicates that a project is amongst the top 10% of the most actively developed projects that we are tracking.

mmdetection

Posts with mentions or reviews of mmdetection. We have used some of these posts to build our list of alternatives and similar projects. The last one was on 2023-04-12.
  • Semantic segementation
    2 projects | /r/computervision | 12 Apr 2023
    When I look for benchmarks I always start here https://paperswithcode.com/task/instance-segmentation/codeless it has the lists of datasets to measure models accross lots o papers. Many are very specific models with low support or community but it gives you a good idea of โ€‹โ€‹the state of the art. It also lists repositories related to good community. https://github.com/open-mmlab/mmdetection seems very active and the one that is being used the most, you could use the models that it has integrated in its model zoo, within the same repository. It has the benchmarks to compare those same models and some of them are from 2022
  • How to Convert Model Mask into Polygon and save JSON?
    1 project | /r/deeplearning | 18 Jan 2023
    MODEL: https://github.com/open-mmlab/mmdetection
  • Object Detection Model for Custom Dataset Training?
    1 project | /r/learnmachinelearning | 11 Jan 2023
    Would it make sense to work with OpenMMLab (https://github.com/open-mmlab/mmdetection) or Pytorch-image-models (https://github.com/rwightman/pytorch-image-models#models) since they offer a variety of models?
  • [P] Image search with localization and open-vocabulary reranking.
    8 projects | /r/MachineLearning | 15 Dec 2022
    I wanted to have a few choices getting localization into image search (index and search time). I immediately thought of using a region proposal network (rpn) from mask-rcnn to create patches that can also be indexed and searched (and add the localisation). I figured it might be somewhat agnostic to classes. I did not want to use mmdetection or detectron2 due to their dependencies and just getting the rpn was not worth it. I was encouraged by the PyTorch native implementations of detection/segmentation models but ended up finding yolox the best.
  • MMDeploy: Deploy All the Algorithms of OpenMMLab
    22 projects | /r/u_Allent_pjlab | 21 Nov 2022
    MMDetection: OpenMMLab detection toolbox and benchmark.
  • Removing the bounding box generated by OnnxRuntime segmentation
    2 projects | /r/computervision | 4 Nov 2022
    I have a semantic segmentation model trained using the mmdetection repo. Then it is converted to the ONNX format using the mmdeploy repo.
  • Keras vs Tensorflow vs Pytorch for a Final year Project
    2 projects | /r/tensorflow | 10 Oct 2022
    E.g. If you consider it an object detection problem it is: detect and localise all the pedestrians in a frame, and classify them by their (intended) action. IMO the easiest way to do this would be with mmdetection, which is built on top of pytorch. Just label your dataset, build a config, and boom you have a model. Inference with that model in only a few lines of code, you won't really need to learn too much to get started.
  • DeepSort with PyTorch(support yolo series)
    13 projects | /r/u_No_Experience9104 | 20 Sep 2022
    MMDetection
  • [D] Pre-trained networks and batch normalization
    1 project | /r/MachineLearning | 15 Sep 2022
    For example, in mmdetection, they expose options in their config & implementation to freeze batch norm layers in backbones and in this config, norm_eval is set to True meaning to freeze tracking of batch norm stats, while the ResNet backbone is frozen up to the 1st stage. Example of their backbone implementation can be found here.
  • Config files in plain Python
    3 projects | /r/Python | 25 Aug 2022
    MMDetection uses config Python scripting. It's easier to define nn.Module objects other than writing class name in a json config file

pytorch-image-models

Posts with mentions or reviews of pytorch-image-models. We have used some of these posts to build our list of alternatives and similar projects. The last one was on 2024-03-18.
  • FLaNK AI Weekly 18 March 2024
    39 projects | dev.to | 18 Mar 2024
  • [D] Hugging face and Timm
    1 project | /r/MachineLearning | 25 Nov 2023
    I am a PyTorch user I work in CV, I usually use the PyTorch models. However, I see people use timm in research papers to train their models I don't understand what it is timm is it a new framework like PyTorch? Further, when I click https://pypi.org/project/timm/ homepage it takes me to hugging face GitHub https://github.com/huggingface/pytorch-image-models is there any connection between timm and hugging face many of my friends use hugging face but I also don't know about hugging face I use simple PyTorch and torchvision.models.
  • FLaNK Stack Weekly for 07August2023
    27 projects | dev.to | 7 Aug 2023
    https://github.com/huggingface/pytorch-image-models https://huggingface.co/docs/timm/index
  • [R] Nvidia RTX 4090 ML benchmarks. Under QEMU/KVM. Image + Transformers. FP16/FP32.
    3 projects | /r/MachineLearning | 14 Jul 2023
    pytorch-image-models
  • Inference on resent, cant work out the problem?
    1 project | /r/MLQuestions | 11 May 2023
    additionally, you might find the timm library handy for this sort of work.
  • Swin Transformer: Hierarchical Vision Transformer Using Shifted Windows
    2 projects | news.ycombinator.com | 9 Apr 2023
    This is still being pursued. Ross Wightmann's timm[0,1] package (now on Hugging Face) has done a lot of this. There's also a V2 of ConvNext[2]. Ross does write about this a lot on Twitter fwiw. I should also mention that there are still many transformer based networks that still beat convs. So there probably won't be a resurgence in convs until someone can show that there's a really strong reason for them. They have some advantages but they also might not be flexible enough for the long range tasks in segmentation and detection. But maybe they are.

    FAIR definitely did great work with ConvNext, and I do hope to see more. There always needs to be people pushing unpopular paradigms.

    [0] https://github.com/huggingface/pytorch-image-models

    [1] https://arxiv.org/abs/2110.00476

    [2] https://arxiv.org/abs/2301.00808

  • Problems with Learning Rate Finder in Pytorch Lightning
    1 project | /r/learnmachinelearning | 2 Mar 2023
    I am doing Binary classification with a pre-trained EfficientNet tf_efficientnet_l2. I froze all weights during training and replaced the classifier with a custom trainable one that looks like:
  • PyTorch at the Edge: Deploying Over 964 TIMM Models on Android with TorchScript and Flutter
    2 projects | /r/SideProject | 15 Feb 2023
    In this post, Iโ€™m going to show you how you can pick from over 900+ SOTA models on TIMM, train them using best practices with Fastai, and deploy them on Android using Flutter.
  • ImageNet Advise
    1 project | /r/deeplearning | 26 Jan 2023
    The other thing is, try to find tricks to speed up your experiments (if not having done so already). The most obvious are mixed precision training, have your model train on a lower resolution input first and then increase the resolution later in the training, stochastic depth, and a bunch more stuffs. Look for implementations in https://github.com/rwightman/pytorch-image-models .
  • Doubt about transformers
    3 projects | /r/MLQuestions | 26 Dec 2022

What are some alternatives?

When comparing mmdetection and pytorch-image-models you can also consider the following projects:

detectron2 - Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.

yolov5 - YOLOv5 ๐Ÿš€ in PyTorch > ONNX > CoreML > TFLite

pytorch-lightning - Build high-performance AI models with PyTorch Lightning (organized PyTorch). Deploy models with Lightning Apps (organized Python to build end-to-end ML systems). [Moved to: https://github.com/Lightning-AI/lightning]

mmcv - OpenMMLab Computer Vision Foundation

PaddleDetection - Object Detection toolkit based on PaddlePaddle. It supports object detection, instance segmentation, multiple object tracking and real-time multi-person keypoint detection.

segmentation_models.pytorch - Segmentation models with pretrained backbones. PyTorch.

mmdetection3d - OpenMMLab's next-generation platform for general 3D object detection.

yolact - A simple, fully convolutional model for real-time instance segmentation.

sahi - Framework agnostic sliced/tiled inference + interactive ui + error analysis plots

tensorflow-image-models - TensorFlow port of PyTorch Image Models (timm) - image models with pretrained weights.