没有名为“自定义\u转换”的模块

Question

我正在做一个项目（时空视频分割），我正在使用COSNET Model written in pytorch。尝试在google colab上运行模型时（特别是下面的部分），我得到了一个错误：

No module named custom_transforms

我试过!pip install custom_transforms但是我得到了：

ERROR: Could not find a version that satisfies the requirement custom_transforms (from versions: none) ERROR: No matching distribution found for custom_transforms

if __name__ == '__main__':
    import custom_transforms as tr
    import torch
    from torchvision import transforms
    from matplotlib import pyplot as plt

    transforms = transforms.Compose([tr.RandomHorizontalFlip(), tr.Resize(scales=[0.5, 0.8, 1]), tr.ToTensor()])

Answer 1

首先，试试这个：

!pip install torchvision
import torchvision.transforms as tr

如果这样做不起作用，似乎这就是您需要的文件：

https://github.com/jfzhang95/DeepGrabCut-PyTorch/blob/master/dataloaders/custom_transforms.py

所以你下载并导入它。你知道吗