1. 程式人生 > >DOTA資料集Faster-RCNN模型問題--resnet-101問題

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沒有對應問題