mirror of
https://github.com/wassname/cr-dv3.git
synced 2026-09-12 12:21:56 +08:00
49 lines
1.4 KiB
Bash
Executable File
49 lines
1.4 KiB
Bash
Executable File
#!/bin/sh
|
|
set -eu
|
|
|
|
# Dependencies
|
|
apt-get update && apt-get install -y \
|
|
build-essential curl freeglut3 gettext git libffi-dev libglu1-mesa \
|
|
libglu1-mesa-dev libjpeg-dev liblua5.1-0-dev libosmesa6-dev \
|
|
libsdl2-dev lua5.1 pkg-config python-setuptools python3-dev \
|
|
software-properties-common unzip zip zlib1g-dev g++
|
|
pip3 install numpy
|
|
|
|
# Bazel
|
|
apt-get install -y apt-transport-https curl gnupg
|
|
curl -fsSL https://bazel.build/bazel-release.pub.gpg | gpg --dearmor > bazel.gpg
|
|
mv bazel.gpg /etc/apt/trusted.gpg.d/
|
|
echo "deb [arch=amd64] https://storage.googleapis.com/bazel-apt stable jdk1.8" | tee /etc/apt/sources.list.d/bazel.list
|
|
apt-get update && apt-get install -y bazel
|
|
|
|
# Build
|
|
git clone https://github.com/deepmind/lab.git
|
|
cd lab
|
|
echo 'build --cxxopt=-std=c++17' > .bazelrc
|
|
bazel build -c opt //python/pip_package:build_pip_package
|
|
./bazel-bin/python/pip_package/build_pip_package /tmp/dmlab_pkg
|
|
pip3 install --force-reinstall /tmp/dmlab_pkg/deepmind_lab-*.whl
|
|
cd ..
|
|
rm -rf lab
|
|
|
|
# Dataset
|
|
mkdir dmlab_data
|
|
cd dmlab_data
|
|
pip3 install Pillow
|
|
curl https://bradylab.ucsd.edu/stimuli/ObjectsAll.zip -o ObjectsAll.zip
|
|
unzip ObjectsAll.zip
|
|
cd OBJECTSALL
|
|
python3 << EOM
|
|
import os
|
|
from PIL import Image
|
|
files = [f for f in os.listdir('.') if f.lower().endswith('jpg')]
|
|
for i, file in enumerate(sorted(files)):
|
|
print(file)
|
|
im = Image.open(file)
|
|
im.save('../%04d.png' % (i+1))
|
|
EOM
|
|
cd ..
|
|
rm -rf __MACOSX OBJECTSALL ObjectsAll.zip
|
|
|
|
apt-get clean
|