關於mask_rcnn 實現的幾點說明

感覺GitHub上Star數最多的竟然不是官方的實現,估計是官方剛開始放出的是基於caffe2 實現的,而當時TensorFlow和keras 大火,所以大家都在尋找基於TensorFlow 和keras 的實現了。後來Facebook官方又放出了基於pytorch的開源實現。確實不錯,速度和精度都有很大的提升,之前詳細瞭解了faster rcnn 的一種pytorch實現的版本,就是借鑑了官方pytorch的版本的一些代碼。

在win10 上進行配置的時候,需要修改幾個cuda 文件。主要的問題是在win10 的pytorch 的API中可能有一兩個函數 沒有實現。造成編譯錯誤。修改之後,完美編譯並且運行。

官網的demo是跑通了,接下來,看下訓練和測試的情況怎麼樣。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章