Merge branch 'master' of https://github.com/ActiveIntelligentSystemsLab/pytorch_enet_ros

Author: ShigemichiMatsuzaki

CMakeLists.txt

@@ -5,7 +5,8 @@ project(pytorch_enet_ros)
 add_compile_options(-std=c++11)
 
 # Locate the cmake file of torchlib
-set(Torch_DIR "/opt/pytorch/pytorch/torch/share/cmake/Torch/")
+#set(Torch_DIR "/opt/pytorch/pytorch/torch/share/cmake/Torch/")
+set(Torch_DIR "$ENV{TORCH_ROOT}/torch/share/cmake/Torch/")
 
 ## Find catkin macros and libraries
 ## if COMPONENTS list like find_package(catkin REQUIRED COMPONENTS xyz)

README.md

@@ -1 +1,4 @@
 # pytorch_enet_ros
+
+A Docker environment for both training network on PyTorch, and inference using ROS is [here](https://github.com/ActiveIntelligentSystemsLab/pytorch-enet-docker).
+This package is only tested on the virtual environment.