Adding new dataset - Githubissues

longcw / faster_rcnn_pytorch

Faster RCNN with PyTorch

MIT License

1.71k stars 466 forks source link

Adding new dataset #35

Closed Rizhiy closed 7 years ago

Rizhiy commented 7 years ago

How should I add new dataset to train on?

I am confused by all the functions in each python file. Which ones do I need to replace?

Rizhiy commented 7 years ago

In particular I'm getting empty overlaps in faster_rcnn/rpn_msr/anchor_target_layer.py:144. A lot of my training images don't contain any objects, can this be a problem?

Rizhiy commented 7 years ago

I found a handler in another repo, which seems to work. I needed Caltech pedestrian handler, check my fork for more info.