DOTA数据集Faster-RCNN模型问题--resnet-101问题

运行之后出现以下问题:

关于resnet-101的问题

修改过程:

rcnn_data_e2e.py文件:

os.environ['PYTHONUNBUFFERED'] = '1'
os.environ['MXNET_CUDNN_AUTOTUNE_DEFAULT'] = '0'
os.environ['MXNET_ENABLE_GPU_P2P'] = '0'

修改为:

os.environ['PYTHONUNBUFFERED'] = '1'
os.environ['MXNET_CUDNN_AUTOTUNE_DEFAULT'] = '1'
os.environ['MXNET_ENABLE_GPU_P2P'] = '1'

在model.py文件里,发现并没有导入.params文件,参数文件不存在问题

将模型前缀名字修改之后,变为resnet_v1_101,结果不变

单步调试,结果到conv1层出现以下情况:

上图是label没有对应问题

猜你喜欢

转载自blog.csdn.net/banxia1995/article/details/84645984
今日推荐