Skip to content

iconbuild/detrex

 
 

Repository files navigation

DETREX

Original repository

The original code is available here.

Installation

The following steps will work on ICON servers (assuming cuda 12.4)

conda create -n detrex python=3.8 -y
conda activate detrex
conda install pytorch torchvision pytorch-cuda=12.4 -c pytorch -c nvidia
python -m pip install -e detectron2
pip install -e .
pip install git+https://github.com/cocodataset/panopticapi.git

About

detrex is a research platform for DETR-based object detection, segmentation, pose estimation and other visual recognition tasks.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages

  • Python 94.0%
  • Cuda 4.0%
  • C++ 1.6%
  • Shell 0.3%
  • Dockerfile 0.1%
  • Makefile 0.0%