Custom Training: Training on custom dataset

ijkguo / mx-rcnn

Parallel Faster R-CNN implementation with MXNet.

Other

669 stars 292 forks source link

Custom Training: Training on custom dataset #36

Closed dubur closed 7 years ago

dubur commented 7 years ago

I have a question about training mx-rcnn on cutstom datasets, does the preparation steps for training is the same to py-faster-rcnn? I went through the io interface of pascal_voc and it didn't contain any rpn_roidb loading functions, did I miss anything?

ijkguo commented 7 years ago

The interface is IMDB, slightly different from py-faster-rcnn. load_rpn_roidb is inherited from IMDB

Ram-Godavarthi commented 6 years ago

@ijkguo What all need to be changed to work on own dataset? I have 2 classes. i have 300 Images of each class.

Can You tell me in details.

Thank You, Any help is really appreciated.