问题标签 [torchvision]

我目前在最终分类层之前使用带有全局平均池（GAP）的 VGG16。使用的 VGG16 模型是 torchvision 提供的模型。

但是，我注意到在 GAP 层之前，有一个 Max Pooling 层。这可以吗，还是应该在 GAP 层之前删除 Max Pooling 层？网络架构如下图所示。

提前致谢。

我最近在我的电脑（操作系统：Windows 10）上安装了 python（3.7.4）和 conda（1.7.2）。

我使用 conda 提示符安装了 pytorch (1.3.1) 和 torchvision (0.4.2)。

我在 conda“环境”选项卡中创建了一个新环境。当我尝试在我的环境中搜索 torchvision 时，我在搜索结果中看不到它（即使我搜索“全部”而不仅仅是“已安装”）。但是，当我切换到“base(root)”环境时，我可以看到 torchvision。

如何将 torchvision 安装到我的环境中？谢谢！

我已经按照他们网站上建议的方式在虚拟环境 (env) 中通过 pip 安装了 Pytorch 和 Torchvision，虽然在我的 python 代码中导入 torchvision 时安装过程中没有发生错误，但会发生以下错误。

我尝试创建新的虚拟环境以从头开始重建，尝试在 conda 环境中通过 conda 安装，但是，这些都没有解决这个问题。
对于任何格式问题，我深表歉意，这是第一次在这里发帖，而且我对使用 python 还比较陌生。
任何帮助将不胜感激。

我的数据，即图像，存储在文件系统中，并ImageFolder通过PyTorch. 因此，training、validation和test数据被手动拆分到文件系统上的不同文件夹中。k-fold cross validation那么，使用时如何申请ImageFolder呢？

我创建了一个 ROS 包，其中有一些 python 脚本。python脚本基于torch模块（它是pytorch模型的推理代码）。当我尝试运行我的脚本时，它给了我一个错误：
ImportError: No module named torch

为了安装 ROS，我使用 了 ROS wiki 的说明。为了验证我的安装，我遵循了 ROS（一个简单的发布者和订阅者）的示例代码，它运行良好。我的系统信息是：

python：3.6.9
torch：1.1.0
torchvision：0.3.0
操作系统内核：Linux 4.15.0-74-通用
操作系统发行版：Ubuntu 18.04.3

我想导入以下库：

我的 CMake 文件如下：

现在，我应该如何编辑 CMAKE 文件以将提到的库添加到我的 ROS 包中？

我正在用 C++ 运行一个受过 Python3 训练的、torch_script 导出的 CNN。但是，我无法在测试数据上重现结果。

对于相同的数据补丁，分类器以各自语言返回的类标签是相反的。不确定 C++ 实现中是否有一些内存损坏，但我至少能够验证 Python 实现给出了正确的结果。我还要如何调试？现在，我正在尝试比较 conv 内核的权重矩阵，以验证模型在两种实现中是否一致。

检查代码中的最后几行以查看差异。

注意 - 两种语言之间的补丁统计信息是相同的，可能是内存在 C++ 中未对齐？

还有什么我可以检查的吗？

Python 导出脚本：

Python 测试脚本

印刷

C++ 测试

印刷

我正在使用预先训练的网络运行一个演示 Web 应用程序。我遇到的问题是，对于我在网络上测试的每个图像，内存都会增加很多并且从未释放过，所以我必须重新启动 Web 服务器才能从不再使用的旧对象中释放内存。这是我运行的网络：

这是内存中最重要部分积累的代码：

哪里net是别名（函数参数），fcn并且inp是图像输入转换为适当的形式以供网络模型使用。

这适用于使用烧瓶和 wsgi 的 apache Web 服务器。我有很多代码，所以我只放最重要的。我不知道如何解决这个问题，因此如果在此处显示一些额外的代码很重要，我不知道是哪一个。

我做了一些研究并找到了一些“快速”的解决方案（有些在 SO 上），但它们并没有太大帮助。例如，这就是我尝试过的。我分离模型调用结果并将其分配给显式变量，然后使用垃圾清理器：

接着

它没有帮助。我假设驻留对象是在外部模型代码下创建的，我希望看到一些解决方案来使用 nn 模型函数从这些对象中释放内存，但不确定我的方向是否正确。

我还考虑在 python 中使用一些“标准”方法，例如创建单独的线程或类似的方法来更好地控制内存。我不是这里的专家，因此将不胜感激。它还必须与 apache、flask 和 wsgi_mod 一起使用。

我下载了 Resnet18 模型来训练模型。

当我输入

表明

如您所见，它清楚地显示了分类器

但是当我这样做时

它显示一个错误

我不知道我在做什么错误，如果你能帮忙，那就太好了。如果你愿意，我可以提供一些额外的细节。

我有一个具有多个输出的模型，确切地说是 4 个：

我想使用make_dotfrom显示它torchviz：

但我收到以下错误：

显然列表没有grad_fn功能，但根据这个讨论，我可以返回一个输出列表。

我究竟做错了什么？

我在安装 torchvision 时遇到了一个奇怪的问题。我刚刚使用以下 pip 命令安装了 torch/torchvision：

pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html

系统：操作系统：win 7，python 3.6.8，带有 cuda

错误：当我这样做时：导入torchvision

我弹出“python.exe - 系统错误”“程序无法启动，因为您的计算机中缺少 avcodec-58.dll...”

我在另一台 win 机器（10，仅限 cpu）上运行了 torch/torchvision，没有问题。

有人建议安装ffmpeg，但没有帮助。问题是我的另一台 Win PC 从来没有 ffmpeg 和 torchvision 安装在那里正常工作。

有人对如何解决这个问题有建议吗？

什么是avcodec-58.dll？

编辑：原来有两个版本的 ffmpeg 静态和共享。显然chocolatey安装了静态的，而torchvision需要共享的，因为它是包含avcodec-58.dll在内的dll。