Skip to content

CharlesShang/Detectron-PYTORCH

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Detectron-PYTORCH

Play with state-of-the-art pedestrian detectors.

Requirements

  1. pytorch
  2. tensorflow (only tensorboard is used)
  3. tensorboardX
    pip install tensorboardX
    
  4. visdom
    pip install visdom
    

Installation

  1. Compiling libs
    cd ./libs
    make
  2. citypersons and cityscape dataset: download coco and extract zips under ./data/citypersons/, so your dirs look like,
    ./data/
            data/citypersons/{annotations}
            data/citypersons/cityscape
                data/citypersons/cityscape/leftImg8bit/{train|val|test}
                data/citypersons/cityscape/gtFine/{train|val|test}

Create citypersons extended dataset

```bash
    python libs/datasets/citypersons2.py
```

Training and evaluation models

```bash
sh ./exper/citypersons/retinanet/train_test_resnet50_softmax.sh
```

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published