感覺GitHub上Star數最多的竟然不是官方的實現,估計是官方剛開始放出的是基於caffe2 實現的,而當時TensorFlow和keras 大火,所以大家都在尋找基於TensorFlow 和keras 的實現了。後來Facebook官方又放出了基於pytorch的開源實現。確實不錯,速度和精度都有很大的提升,之前詳細瞭解了faster rcnn 的一種pytorch實現的版本,就是借鑑了官方pytorch的版本的一些代碼。
在win10 上進行配置的時候,需要修改幾個cuda 文件。主要的問題是在win10 的pytorch 的API中可能有一兩個函數 沒有實現。造成編譯錯誤。修改之後,完美編譯並且運行。
官網的demo是跑通了,接下來,看下訓練和測試的情況怎麼樣。