在Google Collab中永久安装OpenPose

时间:2020-10-13 19:52:13

标签: openpose

我想永久安装OpenPose文件,这样就不必在每次休息后重新打开collab时都安装它们。

我完成了一些安装代码,但我不知道如何进行所需的修改。

import os
from os.path import exists, join, basename, splitext

git_repo_url = 'https://github.com/CMU-Perceptual-Computing-Lab/openpose.git'
project_name = splitext(basename(git_repo_url))[0]
if not exists(project_name):
  # see: https://github.com/CMU-Perceptual-Computing-Lab/openpose/issues/949
  # install new CMake becaue of CUDA10
  !wget -q https://cmake.org/files/v3.13/cmake-3.13.0-Linux-x86_64.tar.gz
  !tar xfz cmake-3.13.0-Linux-x86_64.tar.gz --strip-components=1 -C /usr/local
  # clone openpose
  !git clone -q --depth 1 $git_repo_url
  !sed -i 's/execute_process(COMMAND git checkout master WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}\/3rdparty\/caffe)/execute_process(COMMAND git checkout f019d0dfe86f49d1140961f8c7dec22130c83154 WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}\/3rdparty\/caffe)/g' openpose/CMakeLists.txt
  # install system dependencies
  !apt-get -qq install -y libatlas-base-dev libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler libgflags-dev libgoogle-glog-dev liblmdb-dev opencl-headers ocl-icd-opencl-dev libviennacl-dev
  # install python dependencies
  !pip install -q youtube-dl
  # build openpose
  !cd openpose && rm -rf build || true && mkdir build && cd build && cmake .. && make -j`nproc`
  
from IPython.display import YouTubeVideo

有人可以帮我解决这个问题吗?

1 个答案:

答案 0 :(得分:0)

当你说你想永久安装 openpose 时,我假设你的意思是你想将它安装到你的谷歌驱动器上,而不是每次在 colab 中运行上面的代码时将它安装到临时文件中。

要在您的谷歌驱动器上安装 openpose,而不是在临时 colab 存储上:

(1) 安装您的 gdrive。在您在上述问题中发布的代码块之前添加此代码块。

#Connect your google gdrive
from google.colab import drive
drive.mount('/content/drive')

(2) 将目录更改为您的 gdrive。 在您导入依赖项之后但在代码块的第一行之前,将以下行添加到您的代码中。

#Change the drive to your mounted gdrive
%cd /content/drive/MyDrive

这应该在您的永久 gdrive 上安装 openpose,以便您将来可以从此位置调用 openpose。