如何从 Google Colab 中的 Github 存储库导入自定义模块？

Question

我了解如何在 Colab 中运行单个笔记本。但是，我不确定如何使用存储库中的所有文件，即能够在 Colab 笔记本中导入函数？谢谢你。

Answer 1

假设我们要运行位于存储库“qnn-inference-examples”中的 ipynb 文件，该文件名为“1-complete-connected-binarized-mnist”。

\n

https://github.com/maltanar/qnn-inference-examples

\n

感兴趣的笔记本使用该存储库中定制创建的 QNN 库和函数。是的，我们需要导入该函数。为此，我们应该首先将 repo 文件夹上传到 Google Colab，然后更正/修改库和文件路径。

\n

0)在 Colab 上打开 ipynb 文件“1-complete-connected-binarized-mnist”。如果您愿意，可以重命名它。\n尝试运行它，但可能会出现一些错误（就像我所做的那样）。那么让我们解决这些问题

\n

1)在笔记本顶部插入一个新的代码单元。并在您的 Colab 上克隆存储库：

\n

!git clone https://github.com/maltanar/qnn-inference-examples.git\n

\n

现在，在“content”文件夹下创建新文件夹“qnn-inference-examples”。你应该在左侧看到类似这样的东西。并记住路径“/content/qnn-inference-examples”

\n

\n

2）现在在顶部添加第二个新单元格：

\n

import sys\nsys.path.insert(0,\'/content/qnn-inference-examples\')\n

\n

这将解决尝试导入 QNN 库时无法找到库位置的问题。

\n

3）根据新路径手动修复现有代码上的文件链接。因为库和文件现在存在于文件夹“/content/qnn-inference-examples”下：

\n

例如替换：

\n

img = Image.open("7.png")\n

\n

和

\n

img = Image.open("/content/qnn-inference-examples/7.png")\n

\n

这些步骤应该可以完成工作

\n\n

请注意：这不是我自己的解决方案，而是 2 或 3 个解决方案的混合。感谢 H\xc3\xbcseyin El\xc3\xa7i、KDnuggets 和 Alexandr Haymin

\n

https://medium.com/analytics-vidhya/importing-your-own-python-module-or-python-file-into-colab-3e365f0a35ec

\n

https://www.kdnuggets.com/2018/02/google-colab-free-gpu-tutorial-tensorflow-keras-pytorch.html/2

\n

Answer 2

请看下面的例子：

!git clone https://www.github.com/matterport/Mask_RCNN.git

from google.colab import files

files.os.chdir('Mask_RCNN')

# To find local version of the library
sys.path.append(os.path.join(ROOT_DIR, 'Mask_RCNN'))  

# here is your import
from mrcnn.config import Config