fpn.pytorch
fpn.pytorch copied to clipboard
jwyang
Pytorch implementation of Feature Pyramid Network (FPN) for Object Detection
Hi, I use multi gpus to train but not work. Add option --cuda --mGPUs, I am sure that there are multi gpus to be available. Can you help me ?
Hi, thanks for your implementation. Have you got the results from COCO yet? And I'm still wondering why its performance is worse than your faster-rcnn implementation?
no module named cython_bbox in model.utils.cython It should be `from model.rpn.bbox_transform import bbox_operlaps`
Hi, I am encountered with a problem. Here is the relevant feedback : Called with args: Namespace(batch_size=4, checkepoch=1, checkpoint=0, checkpoint_interval=10000, checksession=1, class_agnostic=False, cuda=True, dataset='pascal_voc', disp_interval=100, lr=0.01, lr_decay_gamma=0.1, lr_decay_step=10, lscale=False, mGPUs=True,...
Hi, Thanks for sharing. I found a line in fpn.py. "roi_level = torch.log(torch.sqrt(h * w) / 224.0)" https://github.com/jwyang/fpn.pytorch/blob/23bd1d2fa09fbb9453f11625d758a61b9d600942/lib/model/fpn/fpn.py#L110 It seems torch.log is based on the e rather than 2 in...
Can I convert this model into trt? Please guide me how can I do this? Thanks
grid_xy = _affine_grid_gen(rois, base_feat.size()[2:], self.grid_size) NameError: global name 'base_feat' is not defined how to solve this problem？Is there someone meet this problem？Hope your request，thanks！
when I train the FPN network on my own dataset for several steps, it goes into the following error. `Traceback (most recent call last): File "trainval_net.py", line 335, in roi_labels...
when I run demo, There will be ImportError: No module named faster_rcnn.faster_rcnn_cascade. there is no faster_rcnn package.
i found rpn_loss_box has always been zero, and that because bbox_outside_weights is 0. maybe i should change positive_weights = 1.0 / num_examples negative_weights = 1.0 / num_examples to positive_weights =...
