pytorch框架下fasterRCNN海上目標自定義數據集訓練測試

代碼鏈接:https://github.com/bubbliiiing/faster-rcnn-pytorch 本地環境如軟件版本: win10 anaconda5 pytorch1.1 1 依賴庫安裝: No module named 'cupy'  解決辦法:因爲本地是cuda9.0環境,所以pip install cupy-cuda90,cuda10則爲cupy-cuda100; 2.數據集
相關文章
相關標籤/搜索