Skip to content

Instantly share code, notes, and snippets.

@Mooan
Created February 4, 2018 23:17
Show Gist options
  • Save Mooan/4a1966c31937c8f201e361d336073df5 to your computer and use it in GitHub Desktop.
Save Mooan/4a1966c31937c8f201e361d336073df5 to your computer and use it in GitHub Desktop.
Jetson TX1 setup for caffe2
## Install Caffe2
https://caffe2.ai/docs/getting-started.html?platform=tegra&configuration=compile
$sudo apt-get install libatlas-base-dev
$cd build
$sudo make install
## Install Detectron
https://github.com/facebookresearch/Detectron/blob/master/INSTALL.md
$ pip2 install future
$ pip2 install numpy>=1.13 pyyaml>=3.12 matplotlib setuptools Cython mock scipy
export PYTHONPATH=$HOME/caffe2/build/
export COCOAPI=$HOME/cocoapi
## Install OpenCV2.
> http://dev.t7.ai/jetson/opencv/
cmake -D WITH_CUDA=ON -D CUDA_ARCH_BIN="5.3" -D CUDA_ARCH_PTX="" -D WITH_OPENGL=ON -D WITH_LIBV4L=ON -D CMAKE_BUILD_TYPE=RELEASE -D CMAKE_INSTALL_PREFIX=/usr/local \
-D WITH_OPENGL=ON \
-D WITH_LIBV4L=ON \
-D CUDA_FAST_MATH=ON \
-D ENABLE_PRECOMPILED_HEADERS=OFF \
-D ENABLE_NEON=ON \
-D BUILD_TESTS=OFF \
-D BUILD_PERF_TESTS=OFF \
-D BUILD_EXAMPLES=OFF \
..
https://devtalk.nvidia.com/default/topic/1007290/jetson-tx2/building-opencv-with-opengl-support-/post/5141945/
pip2 install opencv-python>=3.2
python2 tools/infer_simple.py \
--cfg configs/12_2017_baselines/e2e_mask_rcnn_R-50-FPN_2x.yaml \
--output-dir /tmp/detectron-visualizations \
--image-ext jpg \
--wts https://s3-us-west-2.amazonaws.com/detectron/35861858/12_2017_baselines/e2e_mask_rcnn_R-50-FPN_2x.yaml.02_32_51.SgT4y1cO/output/train/coco_2014_train:coco_2014_valminusminival/generalized_rcnn/model_final.pkl \
demo
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment