py-R-FCN
py-R-FCN copied to clipboard
In demo_rfcn.py, if cfg.TEST.HAS_RPN is set to False, then how do I pass value for blob['rois'] in the _get_blobs function in /py-R-FCN/lib/fast_rcnn/test.py ?
I understand that blobs['rois'] is being set to None. However if None is sent to _get_rois_blob (in the if branch), then another function '_project_im_rois' is giving this (AttributeError: 'NoneType' object has no attribute 'astype') error.
def _get_blobs(im, rois): """Convert an image and RoIs within that image into network inputs.""" blobs = {'data' : None, 'rois' : None}
blobs['data'], im_scale_factors = _get_image_blob(im)
if not cfg.TEST.HAS_RPN:
blobs['rois'] = _get_rois_blob(rois, im_scale_factors)
print blobs['rois'] , im_scale_factors,"<<<<<<<<<<<"
return blobs, im_scale_factors
Hello, have you solved this problem??