fpn.pytorch
fpn.pytorch copied to clipboard
jwyang
Pytorch implementation of Feature Pyramid Network (FPN) for Object Detection
/home/scw4750/anaconda3/envs/pyt0.4/bin/python /media/scw4750/个人文件/liujiajia/fpn.pytorch-master/trainval_net.py Called with args: Namespace(batch_size=1, checkepoch=1, checkpoint=0, checkpoint_interval=10000, checksession=1, class_agnostic=False, cuda='True', dataset='pascal_voc', disp_interval=100, lr=0.001, lr_decay_gamma=0.1, lr_decay_step=5, lscale=False, mGPUs='2', max_epochs=20, net='res101', num_workers=0, optimizer='sgd', resume=False, save_dir='/liujiajia/models', session=1, start_epoch=1, use_tfboard=False) Using config:...
Traceback (most recent call last): File "build.py", line 35, in ffi.build() File "/home/wrc/anaconda3/envs/pytorch0.4/lib/python3.6/site-packages/torch/utils/ffi/__init__.py", line 189, in build _build_extension(ffi, cffi_wrapper_name, target_dir, verbose) File "/home/wrc/anaconda3/envs/pytorch0.4/lib/python3.6/site-packages/torch/utils/ffi/__init__.py", line 111, in _build_extension outfile = ffi.compile(tmpdir=tmpdir,...
Hi @jwyang, I don't think the demo.py file is updated for FPN. Do you have an updated version which you can push? Please do that if you do, it will...
Traceback (most recent call last): File "trainval_net.py", line 330, in roi_labels = FPN(im_data, im_info, gt_boxes, num_boxes) File "/usr/local/lib/python2.7/dist-packages/torch/nn/modules/module.py", line 491, in __call__ result = self.forward(*input, **kwargs) File "/home/pci/save/Technology/Part-B/FPN/lib/model/fpn/fpn.py", line 187,...
When I run trainval.py it reports error: no file or directory: RCNN_rpn and neither did I find the package.
If you are using pytorch 0.4, pls specify dim when softmax function is called in fpn.py and rpn_fpn.py, otherwise you will probably get mAP=0 for every category.
Hi, thanks for your implementation. Why is it worse than Faster-rcnn?
hi,I'm a BUPT Master 1 student,I am study in image process,would you like to let me join you to do some code task ?I will be much greatful for this.
[session 1][epoch 1][iter 2100] loss: 1.2515, lr: 1.00e-03 fg/bg=(32/96), time cost: 46.959169 rpn_cls: 0.0647, rpn_box: 0.0156, rcnn_cls: 0.7545, rcnn_box 0.4920 [session 1][epoch 1][iter 2200] loss: 1.3776, lr: 1.00e-03 fg/bg=(32/96), time...
I tried to change the backbone stride in resnet stage5 to improve the performance, but it didn't work
