Home

Awesome

Open Vocabulary Object Detection with Proposal Mining and Prediction Equalization

Installation

conda create --name detic python=3.8 -y
conda activate detic
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch-lts -c nvidia

# under your working directory
git clone git@github.com:facebookresearch/detectron2.git
cd detectron2
pip install -e .

cd ..
pip install -r requirements.txt

Benchmark evaluation and training

python train_net.py --config-file ./configs/MEDet_COCO_CLIP_R50.yaml --num-gpus $GPUS$
python train_net.py --config-file ./configs/MEDet_COCO_CLIP_R50.yaml --eval-only --num-gpus $GPUS$

Acknowledgement

Our code is developed from Detic