Question

我正在尝试从Github运行此neural style transfer模型，但要在Google Colab上运行，因为我的计算机没有足够的内存/ CPU /任何东西。

我已经将Google驱动器安装到笔记本上，按照this tutorial将模型下载到驱动器文件夹中，将存储库克隆到驱动器中，并且为了测试它是否有效，我使用的是最基本的示例使用以下内容在自述文件中运行Brad Pitt示例的过程：

th neural_style.lua -style_image examples/inputs/picasso_selfport1907.jpg -content_image examples/inputs/brad_pitt.jpg

但是由于某种原因，它将无法正常工作。

使用subprocess.run()刚返回了No such file or directory
使用!th neural_style.lua...返回th: command not found

我已经尝试了四件事，它们都给了我以上两个错误消息的一种变体。有什么想法吗？

以下是完整的笔记本代码，从头到尾/从错误开始再现我在Colab上的设置：

# Mount the drive
from google.colab import drive
drive.mount('/content/drive')

# Clone the repo onto the drive
!git clone https://github.com/jcjohnson/neural-style

# Install Pytorch
!pip3 install http://download.pytorch.org/whl/cu80/torch-0.3.0.post4-cp36-cp36m-linux_x86_64.whl 
!pip3 install torchvision

# Download the models per the github repo instructions
!bash models/download_models.sh

!th neural_style.lua -style_image examples/inputs/picasso_selfport1907.jpg -content_image examples/inputs/brad_pitt.jpg

Answer 1

第一：

%cd /content/
!git clone https://github.com/nagadomi/distro.git torch --recursive
import os
os.chdir('./torch/')
!bash install-deps
!./install.sh
!. ./install/bin/torch-activate

现在，它应该使用指向th的绝对路径来工作：

!/content/torch/install/bin/th neural_style.lua -style_image examples/inputs/picasso_selfport1907.jpg -content_image examples/inputs/brad_pitt.jpg

如何从Google Colab运行lua ML模型

1 个答案: