无法在Google Colab中使用ssd.pytorch包

Question

我正尝试在Google colab中使用此软件包ssd.pytorch，但无法正常工作。我在帖子的最后一行中添加了代码图像。这是我的代码-

import torch
from torch.autograd import Variable
import cv2 
!git clone https://github.com/amdegroot/ssd.pytorch.git #cloning the package in colab first
!pip install -q BaseTransform
from data import BaseTransform, VOC_CLASSES as labelmap 

每当我尝试安装BaseTransform时，都会显示此消息。

  

找不到满足BaseTransform要求的版本（来自版本：）   找不到与BaseTransform匹配的分布

所以我什么也不能导入。我应该安装我首先克隆的东西吗？但是，当我尝试安装ssd.pytorch时，会显示相同的消息。

!pip install -q ssd.pytorch

  

找不到满足ssd.pytorch要求的版本（来自版本：）   找不到与ssd.pytorch匹配的分布

从文件夹数据导入时，我应该使用该文件夹名称进行安装吗？

!pip install -q data

出于某种原因，这行得通，我不知道为什么。但是当我使用此行时仍然无法导入任何内容。

from data import BaseTransform, VOC_CLASSES as labelmap

  

ImportError跟踪（最近一次通话）    在（）中   ----> 1从数据导入BaseTransform，VOC_CLASSES作为labelmap。 ImportError：无法导入名称“ BaseTransform”

     

注意：如果由于缺少软件包而导致导入失败，则可以   使用！pip或！apt手动安装依赖项。要查看安装一些常见依赖项的示例，请单击   下面的“打开示例”按钮。

我应该怎么做才能正确导入软件包？ 这是colab中的代码：

Answer 1

您遇到的问题是，在导入软件包之前您没有更改目录。

在导入之前只需运行以下代码

import os

os.chdir('ssd.pytorch')

此外，BaseTransform是ssd.pytorch / data中定义的类。您无法安装。