时间戳

首页 > 解决方案 > 没有名为“custom_transforms”的模块

python - 没有名为“custom_transforms”的模块

问题描述

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

没有名为 custom_transforms 的模块

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

错误：找不到满足 custom_transforms 要求的版本（来自版本：无） 错误：未找到 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()])

标签： pythonmoduledeep-learningpytorchimporterror

解决方案

首先，试试这个：

!pip install torchvision
import torchvision.transforms as tr

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

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

所以你下载它并导入它。

推荐阅读