Under development

- Repackage
- ROS Support
- Network Tunings
- Knowledge Distilation (TODO)

.dockerignore

@@ -1,6 +1,23 @@
 ./models
 ./models/*
+./models/*/*
+./models/*/*/*
+./models/*/*/*/*
 models
+models/*
+models/*/*
+models/*/*/*
+models/*/*/*/*
+*.meta
+*.index
+*.data-*
+*.ckpt.*
+
 ./tests
 ./tests/*
-tests
+tests
+graph*.pb
+chk*.meta
+lifting
+slim
+

CMakeLists.txt

@@ -0,0 +1,28 @@
+cmake_minimum_required(VERSION 2.8.3)
+project(tfpose_ros)
+
+## Add support for C++11, supported in ROS Kinetic and newer
+add_definitions(-std=c++11)
+
+find_package(catkin REQUIRED COMPONENTS
+  roscpp
+  rospy
+  std_msgs
+  message_generation
+)
+
+# Generate messages in the 'msg' folder
+add_message_files(
+  FILES
+  BodyPartElm.msg
+  Person.msg
+  Persons.msg
+)
+
+generate_messages(
+  DEPENDENCIES std_msgs
+)
+
+catkin_package(
+  CATKIN_DEPENDS message_generation message_runtime
+)

Dockerfile

@@ -1,38 +1,8 @@
-FROM ubuntu:16.04
+FROM idock.daumkakao.io/kakaobrain/deepcloud-sshd:openpose-preprocess
 
-ENV http_proxy=http://10.41.249.28:8080 https_proxy=http://10.41.249.28:8080
-
-RUN apt-get update -yq && apt-get install -yq build-essential cmake git pkg-config wget zip && \
-apt-get install -yq libjpeg8-dev libtiff5-dev libjasper-dev libpng12-dev && \
-apt-get install -yq libavcodec-dev libavformat-dev libswscale-dev libv4l-dev && \
-apt-get install -yq libgtk2.0-dev && \
-apt-get install -yq libatlas-base-dev gfortran && \
-apt-get install -yq python3 python3-dev python3-pip python3-setuptools python3-tk git && \
-pip3 install numpy && \
-cd ~ && git clone https://github.com/Itseez/opencv.git && \
-cd opencv && mkdir build && cd build && \
-cmake -D CMAKE_BUILD_TYPE=RELEASE \
-	-D CMAKE_INSTALL_PREFIX=/usr/local \
-	-D INSTALL_PYTHON_EXAMPLES=ON \
-	-D BUILD_opencv_python3=yes -D PYTHON_EXECUTABLE=/usr/bin/python3 .. && \
-make -j8 && make install && rm -rf /root/opencv/ && \
-mkdir -p /root/tf-openpose && \
-rm -rf /tmp/*.tar.gz && \
-apt-get clean && rm -rf /tmp/* /var/tmp* /var/lib/apt/lists/* && \
-rm -f /etc/ssh/ssh_host_* && rm -rf /usr/share/man?? /usr/share/man/??_*
-
-COPY . /root/tf-openpose/
+COPY ./*.py /root/tf-openpose/
 WORKDIR /root/tf-openpose/
 
-RUN cd /root/tf-openpose/ && pip3 install -U setuptools && \
-pip3 install tensorflow && pip3 install -r requirements.txt
-
-RUN cd /root && git clone https://github.com/cocodataset/cocoapi && \
-pip3 install cython && \
-cd cocoapi/PythonAPI && python3 setup.py build_ext --inplace && python3 setup.py build_ext install && \
-mkdir /coco && cd /coco && wget http://images.cocodataset.org/annotations/annotations_trainval2017.zip && \
-unzip annotations_trainval2017.zip && rm -rf annotations_trainval2017.zip
+RUN cd /root/tf-openpose/ && pip3 install -r requirements.txt
 
 ENTRYPOINT ["python3", "pose_dataworker.py"]
-
-ENV http_proxy= https_proxy=

README.md

@@ -1,4 +1,4 @@
-# tf-openpose
+# tf-pose-estimation
 
 'Openpose' for human pose estimation have been implemented using Tensorflow. It also provides several variants that have made some changes to the network structure for **real-time processing on the CPU or low-power embedded devices.**
 
@@ -123,6 +123,12 @@ See : [etcs/training.md](./etcs/training.md)
 
 [4] Keras Openpose : https://github.com/michalfaber/keras_Realtime_Multi-Person_Pose_Estimation
 
+### Lifting from the deep
+
+[1] Arxiv Paper : https://arxiv.org/abs/1701.00295
+
+[2] https://github.com/DenisTome/Lifting-from-the-Deep-release
+
 ### Mobilenet
 
 [1] Original Paper : https://arxiv.org/abs/1704.04861