This code includes several extensions we have made to our conference version. Please read the paper for details.
python 2.7, pytorch 0.4.1, numpy, cv2, scipy.
-
Download the source code, the dataset and the pretrained model.
-
Run
TrainModel.py
to train a new model on our dataset or Rundemo_eval.py
to test the pretrained model on any images. -
To change the aspect ratio of generated crops, please change the
generate_bboxes
function incroppingDataset.py
(line 115).
The executable annotation software can be found here.