提交 582c622d 编写于 作者: G gineshidalgo99

Fix Ubuntu script paths

上级 25b54852
from .pycaffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, RMSPropSolver, AdaDeltaSolver, AdamSolver, NCCL, Timer
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
from ._caffe import __version__
from .proto.caffe_pb2 import TRAIN, TEST
from .classifier import Classifier
from .detector import Detector
from . import io
from .net_spec import layers, params, NetSpec, to_proto
......@@ -46,4 +46,4 @@ exitIfError
echo "------------------------- Caffe and OpenPose Installed -------------------------"
echo ""
\ No newline at end of file
echo ""
......@@ -62,11 +62,15 @@ echo ""
echo "------------------------- Compiling OpenPose -------------------------"
# Go back to main folder
cd ..
# Copy Makefile.config
if [[ $ubuntu_le_14 == true ]]; then
cp ubuntu/Makefile.config.Ubuntu14_cuda8.example Makefile.config
else
cp ubuntu/Makefile.config.Ubuntu16_cuda8.example Makefile.config
fi
# Compile OpenPose
make all -j$NUM_CORES
exitIfError
echo "------------------------- OpenPose Compiled -------------------------"
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册