Skip to content

Commit 582c622

Browse files
Fix Ubuntu script paths
1 parent 25b5485 commit 582c622

File tree

3 files changed

+13
-1
lines changed

3 files changed

+13
-1
lines changed
Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
from .pycaffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, RMSPropSolver, AdaDeltaSolver, AdamSolver, NCCL, Timer
2+
from ._caffe import init_log, log, set_mode_cpu, set_mode_gpu, set_device, Layer, get_solver, layer_type_list, set_random_seed, solver_count, set_solver_count, solver_rank, set_solver_rank, set_multiprocess, has_nccl
3+
from ._caffe import __version__
4+
from .proto.caffe_pb2 import TRAIN, TEST
5+
from .classifier import Classifier
6+
from .detector import Detector
7+
from . import io
8+
from .net_spec import layers, params, NetSpec, to_proto

ubuntu/install_caffe_and_openpose_if_cuda8.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,4 +46,4 @@ exitIfError
4646

4747

4848
echo "------------------------- Caffe and OpenPose Installed -------------------------"
49-
echo ""
49+
echo ""

ubuntu/install_openpose_if_cuda8.sh

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -62,11 +62,15 @@ echo ""
6262

6363

6464
echo "------------------------- Compiling OpenPose -------------------------"
65+
# Go back to main folder
66+
cd ..
67+
# Copy Makefile.config
6568
if [[ $ubuntu_le_14 == true ]]; then
6669
cp ubuntu/Makefile.config.Ubuntu14_cuda8.example Makefile.config
6770
else
6871
cp ubuntu/Makefile.config.Ubuntu16_cuda8.example Makefile.config
6972
fi
73+
# Compile OpenPose
7074
make all -j$NUM_CORES
7175
exitIfError
7276
echo "------------------------- OpenPose Compiled -------------------------"

0 commit comments

Comments
 (0)