fpn.pytorch
fpn.pytorch copied to clipboard
jwyang
Pytorch implementation of Feature Pyramid Network (FPN) for Object Detection
when i try to resume to train the model ： RuntimeError: Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for argument #4 'other'
the positive and negative weights are normalized by num_examples in anchor_target_layer_fpn.py. num_examples is calculated based on index i outside of i's batch loop, thus the only number of examples that...
Hi, Please notice you have an inconsistant reference to the order of the anchors (lines 86 and 79) https://github.com/jwyang/fpn.pytorch/blob/23bd1d2fa09fbb9453f11625d758a61b9d600942/lib/model/rpn/rpn_fpn.py#L86 https://github.com/jwyang/fpn.pytorch/blob/23bd1d2fa09fbb9453f11625d758a61b9d600942/lib/model/rpn/rpn_fpn.py#L79 lets say k is the number of anchors than in...
I'm trying to crop the images and then send it into the backbone Resnet, I was wondering where I can find the DataLoader and how it was loaded and how...
Traceback (most recent call last): File "test_net.py", line 329, in imdb.evaluate_detections(all_boxes, output_dir) File "/home/wangzhaowei/pycharm/fpn.pytorch-master/lib/datasets/pascal_voc.py", line 349, in evaluate_detections self._do_python_eval(output_dir) File "/home/wangzhaowei/pycharm/fpn.pytorch-master/lib/datasets/pascal_voc.py", line 312, in _do_python_eval use_07_metric=use_07_metric) File "/home/wangzhaowei/pycharm/fpn.pytorch-master/lib/datasets/voc_eval.py", line 126,...
per time stop train,must be show a new process "irq/133-nvidia",cpu 100%, nvidia net show : systemctl daemon-reload systemctl enable nvidia-persistenced systemctl start nvidia-persistenced
Loading pretrained weights from data/pretrained_model/resnet101_caffe.pth /home/ztd/anaconda2/envs/py2.7_tf1.4_tff/lib/python2.7/site-packages/torch/nn/functional.py:1749: UserWarning: Default upsampling behavior when mode=bilinear is changed to align_corners=False since 0.4.0. Please specify align_corners=True if the old behavior is desired. See the documentation...
Great work! I want to ask which version of pytorch to use. thx!
Beginner in pytorch, I found trouble to understand the code about loading the images into the network and how to resize them.