AI Habitat上Habitat-Lab for Navigation教程在Colab的一些补充

解决方法参考的是 Habitat-sim与Habitat-lab在Colab上的环境配置

Colab Setup and Imports步骤报错ModuleNotFoundError: No module named 'habitat'

运行

!add-apt-repository ppa:ubuntu-toolchain-r/test -y
!apt-get update --fix-missing
!apt-get dist-upgrade

然后重启环境即可。

具体原因和解决方法是如何实现的我也不清楚。

关于Colab的重启环境，直接刷新页面可能会让你在note上编辑的代码丢失，这边参考 colab在更换python包版本时，如何正确重启，运行

try:
  from gem.utils import graph_util, plot_util
except (ImportError, KeyError, ModuleNotFoundError):
  ## code to install gem
  print('Stopping RUNTIME. Colaboratory will restart automatically. Please run again.')
  exit()

即可。

接着继续报错 ModuleNotFoundError: No module named ‘habitat_baselines’

这边原因怀疑是没有自动安装habitat-baselines和habitat-lab依赖导致的，自己收到进入对应目录运行安装脚本即可。

%cd /content/habitat-lab/habitat-baselines
!python setup.py install
%cd /content/habitat-lab/habitat-lab
!python setup.py install

然后重启环境即可。

Setup PointNav Task，运行只生成一个图像

原因是

action = valid_actions.pop()

pop()默认是弹出最后一个元素，直接stop导致，改成

action = valid_actions.pop(0)

即可。