解決Caffe在Pycharm沒有代碼提示的問題

參照上一篇博客完美解決Pytorch在Pycharm沒有代碼提示的問題的方法,通過Python的help命令搞了一個__init__.pyi文件。
直接說明解決方案:

  • 升級Pycharm到版本2019.1.1。
  • 從項目PythonResources下載caffe文件夾裏的__init__.pyi
  • 把下載的__init__.pyi複製到Caffe安裝目錄/python/caffe/裏面。

直接通過help命令查看文檔很不方便,故把文檔保存爲文本文件,方便查看。保存模塊的文檔的代碼爲:

# coding=utf8
import sys
import pydoc


def save_doc(filepath, module):
    f = open(filepath, 'w')
    sys.stdout = f
    pydoc.help(module)
    f.close()
    sys.stdout = sys.__stdout__


if __name__ == '__main__':
    # 保存caffe.Timer的文檔到test.txt
    save_doc('test.txt', 'caffe.Timer')


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