search

ijkguo / mx-rcnn

Parallel Faster R-CNN implementation with MXNet.
Other
669 stars 292 forks source link

train faster rcnn #27

Closed zhyj3038 closed 7 years ago

zhyj3038 commented 7 years ago

hello, when I run "python -m tools.train_alternate", error occurs as follow:

NFO:root:########## train rpn with imagenet init voc_2007_trainval gt roidb loaded from ......... File "/home/mxnet_0724/example/rcnn/tools/train_alternate.py", line 52, in train_rpn args['rpn_conv_3x3_weight'] = mx.random.normal(mean=0, stdvar=0.01, shape=arg_shape_dict['rpn_conv_3x3_weight']) TypeError: normal() got an unexpected keyword argument 'mean'

I find mxnet.random.normal is defined in python/mexnt/random.py file , but the input argument is "loc", not "mean", I just wondor how can you run it withnot change name of this argument???

zhyj3038 commented 7 years ago

sorry，find the latest version have fix it :)