Skip to content

Latest commit

 

History

History
 
 

README.md

VimDet

Envs.

  • Python 3.9.19

    • conda create -n your_env_name python=3.9.19
    # some packages
    pip3 install numpy==1.23.4
    pip3 install iopath==0.1.9
    pip3 install cloudpickle==2.2.1
    pip3 install omegaconf==2.3.0
    pip3 install Pillow==8.2.0
    pip3 install requests==2.28.1
    pip3 install sympy==1.12.1
    pip3 install pycocotools==2.0.6
    pip3 install Shapely==1.8.1
    
    # install d2
    cd det
    pip3 install -e .
    
    # link datasets
    mkdir datasets
    cd datasets
    ln -s /path/to/coco ./
    
  • Meet some problems of env? Requirements for reference: det-requirements.txt

Train Your VimDet

bash scripts/ft_vim_tiny_vimdet.sh

Test Your VimDet

bash scripts/eval_vim_tiny_vimdet.sh


Detectron2 Part

Detectron2 is Facebook AI Research's next generation library that provides state-of-the-art detection and segmentation algorithms. It is the successor of Detectron and maskrcnn-benchmark. It supports a number of computer vision research projects and production applications in Facebook.


Learn More about Detectron2

Explain Like I’m 5: Detectron2 Using Machine Learning with Detectron2
Explain Like I’m 5: Detectron2 Using Machine Learning with Detectron2

What's New

  • Includes new capabilities such as panoptic segmentation, Densepose, Cascade R-CNN, rotated bounding boxes, PointRend, DeepLab, ViTDet, MViTv2 etc.
  • Used as a library to support building research projects on top of it.
  • Models can be exported to TorchScript format or Caffe2 format for deployment.
  • It trains much faster.

See our blog post to see more demos and learn about detectron2.

Installation

See installation instructions.

Getting Started

See Getting Started with Detectron2, and the Colab Notebook to learn about basic usage.

Learn more at our documentation. And see projects/ for some projects that are built on top of detectron2.

Model Zoo and Baselines

We provide a large set of baseline results and trained models available for download in the Detectron2 Model Zoo.

License

Detectron2 is released under the Apache 2.0 license.

Citing Detectron2

If you use Detectron2 in your research or wish to refer to the baseline results published in the Model Zoo, please use the following BibTeX entry.

@misc{wu2019detectron2,
  author =       {Yuxin Wu and Alexander Kirillov and Francisco Massa and
                  Wan-Yen Lo and Ross Girshick},
  title =        {Detectron2},
  howpublished = {\url{https://github.com/facebookresearch/detectron2}},
  year =         {2019}
}