HuangZiliAndy / RPNSD

PyTorch implementation of RPNSD
MIT License
60 stars 15 forks source link

NMS scripts are missing #11

Open lsari opened 4 years ago

lsari commented 4 years ago

I am trying to train a model from scratch but I realized that "from model.nms.nms_wrapper import nms" lines (e.g. in model/rpn/proposal_layer.py) fail as the repo does not include nms directory under scripts/model/. As far as I understand, cluster_nms.py is also not the correct one as it also calls model.nms. Could you please share the code for NMS to fix this issue? Or am I missing something?