tianyolanda / fpn-pytorch1.0 Goto Github PK

when i train model, rpn_cls loss and rpn_box loss are nan. the bg_cnt=0. dose anyone know how to solve it

Hello, I use your demo, the test has been wrong, the error message is as follows:

File "test_net.py", line 236, in
_, _, _, _, _ = fpn(im_data, im_info, gt_boxes, num_boxes)
File "/home/xiaojie/anaconda3/envs/pytorch/lib/python3.6/site-packages/torch/nn/modules/module.py", line 493, in call
result = self.forward(*input, **kwargs)
File "/home/xiaojie/python_program/pytorch/fpn-pytorch1.0/lib/model/fpn/fpn.py", line 224, in forward
roi_pool_feat = self._PyramidRoI_Feat(mrcnn_feature_maps, rois, im_info)
File "/home/xiaojie/python_program/pytorch/fpn-pytorch1.0/lib/model/fpn/fpn.py", line 141, in _PyramidRoI_Feat
return roi_pool_feat
UnboundLocalError: local variable 'roi_pool_feat' referenced before assignment

I don't know why it is wrong, could you help me solve it? Thank you