pytorch-retinanet

pytorch-retinanet copied to clipboard

It will be nice to be able to train using other standard networks as well, since FPN requires too much memory. Right now, changing the base network in retinanet.py (e.g.,...

mubastan