[pretty_dumm_guy]

Hi,

Awesome work. Something similar to what I did, though for semantic segmentation.

A small suggestion: From what I understand, you have hard-coded the Fast-RCNN predictor.

  self._model.roi_heads.box_predictor = FastRCNNPredictor(in_features, len(classes) + 1)

Having the ability to plugin users' own models would be better, I believe. This would translate to making the model class more abstract.

Well done!

[alanbi]

Thanks for the suggestion! That's a feature I definitely plan on adding sometime in the near future.