Register
Login
Resources
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Product
Data Engine LLMs Platform Enterprise
Pricing Explore
Connect to our Discord channel

CMakeLists.txt 41 KB

You have to be logged in to leave a comment. Sign In
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
1001
1002
1003
1004
1005
1006
1007
1008
1009
1010
1011
1012
1013
1014
1015
1016
1017
1018
1019
1020
1021
1022
1023
1024
1025
1026
1027
1028
1029
1030
1031
1032
1033
1034
1035
1036
1037
1038
1039
1040
1041
1042
1043
1044
1045
1046
1047
1048
1049
1050
1051
1052
1053
1054
1055
1056
1057
1058
1059
1060
  1. ### VERSION INFO
  2. set(OpenPose_VERSION_MAJOR 1)
  3. set(OpenPose_VERSION_MINOR 5)
  4. set(OpenPose_VERSION_PATCH 1)
  5. set(OpenPose_VERSION ${OpenPose_VERSION_MAJOR}.${OpenPose_VERSION_MINOR}.${OpenPose_VERSION_PATCH})
  6. ### OS-DEPENDENT FLAGS
  7. set(CMAKE_MACOSX_RPATH 1)
  8. ### CMAKE HEADERS
  9. # Ubuntu 18 default. After 3.8, no need for find_CUDA
  10. # https://cmake.org/cmake/help/v3.10/module/FindCUDA.html
  11. # https://cmake.org/cmake/help/v3.10/command/project.html
  12. # https://devblogs.nvidia.com/building-cuda-applications-cmake/
  13. if (${CMAKE_VERSION} VERSION_GREATER 3.9.0)
  14. cmake_policy(SET CMP0048 NEW)
  15. project(OpenPose VERSION ${OpenPose_VERSION})
  16. # # Not tested
  17. # cmake_policy(SET CMP0048 NEW)
  18. # set(CUDACXX /usr/local/cuda/bin/nvcc)
  19. # project(OpenPose VERSION ${OpenPose_VERSION} LANGUAGES CXX CUDA)
  20. # set(AUTO_FOUND_CUDA TRUE)
  21. # # else
  22. # set(AUTO_FOUND_CUDA FALSE)
  23. # Ubuntu 16 default
  24. elseif (${CMAKE_VERSION} VERSION_GREATER 3.0.0)
  25. cmake_policy(SET CMP0048 NEW)
  26. project(OpenPose VERSION ${OpenPose_VERSION})
  27. else (${CMAKE_VERSION} VERSION_GREATER 3.9.0)
  28. project(OpenPose)
  29. endif (${CMAKE_VERSION} VERSION_GREATER 3.9.0)
  30. cmake_minimum_required(VERSION 3.0 FATAL_ERROR) # required by cmake eigen finder to find locally installed versions (e.g., CUDA 9.2 support was added in eigen 3.3.5)
  31. # cmake_minimum_required(VERSION 2.8.7 FATAL_ERROR) # min. cmake version recommended by Caffe
  32. ### Extra functionality
  33. include(cmake/Utils.cmake)
  34. if (NOT WIN32 AND NOT UNIX AND NOT APPLE)
  35. message(FATAL_ERROR "Unsupported operating system. Only Windows, Mac and Unix systems supported.")
  36. endif (NOT WIN32 AND NOT UNIX AND NOT APPLE)
  37. ### CMAKE_BUILD_TYPE
  38. # Default: Release
  39. # Bug fixed: By default, it uses something different to Release, that provokes OpenPose to be about 15% slower than
  40. # it should be.
  41. # Is CMAKE_BUILD_TYPE "Debug" or "MinSizeRel" or "RelWithDebInfo"?
  42. set(CMAKE_BUILD_TYPE_KNOWN FALSE)
  43. if (${CMAKE_BUILD_TYPE} MATCHES "Debug")
  44. set(CMAKE_BUILD_TYPE_KNOWN TRUE)
  45. endif (${CMAKE_BUILD_TYPE} MATCHES "Debug")
  46. if (${CMAKE_BUILD_TYPE} MATCHES "MinSizeRel")
  47. set(CMAKE_BUILD_TYPE_KNOWN TRUE)
  48. endif (${CMAKE_BUILD_TYPE} MATCHES "MinSizeRel")
  49. if (${CMAKE_BUILD_TYPE} MATCHES "RelWithDebInfo")
  50. set(CMAKE_BUILD_TYPE_KNOWN TRUE)
  51. endif (${CMAKE_BUILD_TYPE} MATCHES "RelWithDebInfo")
  52. # Assign proper CMAKE_BUILD_TYPE
  53. if (${CMAKE_BUILD_TYPE_KNOWN})
  54. set(CMAKE_BUILD_TYPE "Release" CACHE STRING "Choose the type of build.")
  55. else (${CMAKE_BUILD_TYPE_KNOWN})
  56. set(CMAKE_BUILD_TYPE "Release" CACHE STRING "Choose the type of build." FORCE)
  57. endif (${CMAKE_BUILD_TYPE_KNOWN})
  58. set_property(CACHE CMAKE_BUILD_TYPE PROPERTY STRINGS "Release" "Debug" "MinSizeRel" "RelWithDebInfo")
  59. ### FLAGS
  60. if (WIN32)
  61. # TODO -- Makeshift solution -- This prevents rerunning build again
  62. # https://gitlab.kitware.com/cmake/cmake/issues/16783
  63. set(CMAKE_SUPPRESS_REGENERATION ON)
  64. string (REPLACE "/D_WINDOWS" "" CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}")
  65. string (REPLACE "/DWIN32" "" CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}")
  66. # /GL option causes the code to crash -- fix this
  67. # sdl flags causes error -- error : unknown attribute \"guard\"
  68. # https://stackoverflow.com/questions/43697154/detect-current-cmake-version-using-cmake
  69. message(STATUS "CMAKE_VERSION = ${CMAKE_VERSION}")
  70. # Remove "minSizeRel" and "relWithDebInfo"
  71. if (${CMAKE_VERSION} VERSION_LESS "3.14.0")
  72. set(CMAKE_CONFIGURATION_TYPES Release Debug CACHE TYPE INTERNAL FORCE)
  73. else (${CMAKE_VERSION} VERSION_LESS "3.14.0")
  74. # But the above line gives warning on CMake >= 3.14, alternative
  75. set(CMAKE_CONFIGURATION_TYPES Release Debug CACHE STRING "" FORCE)
  76. endif (${CMAKE_VERSION} VERSION_LESS "3.14.0")
  77. set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} /MP")
  78. string(REPLACE "/W3" "/W4" CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}")
  79. string(REPLACE "/GR" "" CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}")
  80. string(REPLACE "/O2" "/Ox" CMAKE_CXX_FLAGS_RELEASE "${CMAKE_CXX_FLAGS_RELEASE}")
  81. string(REPLACE "/Ob2" "" CMAKE_CXX_FLAGS_RELEASE "${CMAKE_CXX_FLAGS_RELEASE}")
  82. set(CMAKE_CXX_FLAGS_RELEASE "${CMAKE_CXX_FLAGS_RELEASE} /Ot /Oi /Gy /Z7")
  83. set(CMAKE_SHARED_LINKER_FLAGS_RELEASE "${CMAKE_SHARED_LINKER_FLAGS_RELEASE} /LTCG:incremental /OPT:REF /OPT:ICF")
  84. string(REPLACE "/MDd" "/MD" CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG}")
  85. string(REPLACE "/Zi" "/Z7" CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG}")
  86. string(REPLACE "/RTC1" "" CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG}")
  87. elseif (UNIX)
  88. # Turn on C++11
  89. add_definitions(-std=c++11)
  90. set(CMAKE_CXX_FLAGS_RELEASE "-O3")
  91. elseif (APPLE)
  92. # Turn on C++11
  93. set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11")
  94. set(CMAKE_CXX_FLAGS_RELEASE "-O3")
  95. endif (WIN32)
  96. # C++ additional flags
  97. if (CMAKE_COMPILER_IS_GNUCXX)
  98. message(STATUS "GCC detected, adding compile flags")
  99. # set(OP_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fopenmp -Wpedantic -Wall -Wextra -Wfatal-errors")
  100. set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fopenmp")
  101. set(OP_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wpedantic -Wall -Wextra -Wfatal-errors")
  102. # set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} -Wpedantic -Wall -Wextra -Wfatal-errors")
  103. endif (CMAKE_COMPILER_IS_GNUCXX)
  104. # Select the Enhanced Instruction Set
  105. set(INSTRUCTION_SET NONE CACHE STRING "Enable Enhanced Instruction Set")
  106. set_property(CACHE INSTRUCTION_SET PROPERTY STRINGS NONE AVX2)
  107. if (${INSTRUCTION_SET} MATCHES "AVX2")
  108. add_definitions("-DWITH_AVX")
  109. endif (${INSTRUCTION_SET} MATCHES "AVX2")
  110. # Windows
  111. if (WIN32)
  112. # Suboptions for Enhanced Instruction Set
  113. if (${INSTRUCTION_SET} MATCHES "AVX2")
  114. add_definitions("/arch:AVX2")
  115. endif (${INSTRUCTION_SET} MATCHES "AVX2")
  116. # Ubuntu
  117. else (WIN32) # if (CMAKE_COMPILER_IS_GNUCXX)
  118. if (${INSTRUCTION_SET} MATCHES "AVX2")
  119. set(SIMD_FLAGS "${SIMD_FLAGS} -mavx -march=native")
  120. endif (${INSTRUCTION_SET} MATCHES "AVX2")
  121. message(STATUS "GCC detected, adding compile flags")
  122. set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${SIMD_FLAGS}")
  123. endif (WIN32)
  124. # if (CMAKE_COMPILER_IS_GNUCXX)
  125. # set_property(CACHE INSTRUCTION_SET PROPERTY STRINGS NONE SSE4 AVX)
  126. # if (${INSTRUCTION_SET} MATCHES "SSE4")
  127. # add_definitions("-DWITH_SSE4")
  128. # set(SIMD_FLAGS "${SIMD_FLAGS} -msse4.1")
  129. # endif (${INSTRUCTION_SET} MATCHES "SSE4")
  130. # if (${INSTRUCTION_SET} MATCHES "AVX2")
  131. # add_definitions("-DWITH_AVX")
  132. # set(SIMD_FLAGS "${SIMD_FLAGS} -mavx -march=native")
  133. # endif (${INSTRUCTION_SET} MATCHES "AVX2")
  134. # message(STATUS "GCC detected, adding compile flags")
  135. # set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${SIMD_FLAGS}")
  136. # endif (CMAKE_COMPILER_IS_GNUCXX)
  137. # if (WIN32)
  138. # # set_property(CACHE INSTRUCTION_SET PROPERTY STRINGS NONE SSE SSE2 AVX AVX2 IA32)
  139. # # Suboptions for Enhanced Instruction Set
  140. # if (${INSTRUCTION_SET} MATCHES "SSE")
  141. # add_definitions("/arch:SSE")
  142. # elseif (${INSTRUCTION_SET} MATCHES "SSE2")
  143. # add_definitions("/arch:SSE2")
  144. # elseif (${INSTRUCTION_SET} MATCHES "AVX")
  145. # add_definitions("/arch:AVX")
  146. # elseif (${INSTRUCTION_SET} MATCHES "AVX2")
  147. # add_definitions("/arch:AVX2")
  148. # elseif (${INSTRUCTION_SET} MATCHES "IA32")
  149. # add_definitions("/arch:IA32")
  150. # endif (${INSTRUCTION_SET} MATCHES "SSE")
  151. # endif (WIN32)
  152. ### PROJECT OPTIONS
  153. # Select the DL Framework
  154. set(DL_FRAMEWORK CAFFE CACHE STRING "Select Deep Learning Framework.")
  155. if (WIN32)
  156. set_property(CACHE DL_FRAMEWORK PROPERTY STRINGS CAFFE)
  157. else (WIN32)
  158. set_property(CACHE DL_FRAMEWORK PROPERTY STRINGS CAFFE NV_CAFFE)
  159. # set_property(CACHE DL_FRAMEWORK PROPERTY STRINGS CAFFE NV_CAFFE PYTORCH TENSORFLOW)
  160. endif (WIN32)
  161. # Suboptions for Caffe DL Framework
  162. include(CMakeDependentOption)
  163. if (${DL_FRAMEWORK} MATCHES "CAFFE" OR ${DL_FRAMEWORK} MATCHES "NV_CAFFE")
  164. CMAKE_DEPENDENT_OPTION(BUILD_CAFFE "Build Caffe as part of OpenPose." ON
  165. "DL_FRAMEWORK" ON)
  166. # OpenPose flags
  167. add_definitions(-DUSE_CAFFE)
  168. # Nvidia NVCaffe
  169. if (${DL_FRAMEWORK} MATCHES "NV_CAFFE")
  170. MESSAGE(STATUS "Using NVIDIA NVCaffe")
  171. add_definitions(-DNV_CAFFE)
  172. endif (${DL_FRAMEWORK} MATCHES "NV_CAFFE")
  173. endif (${DL_FRAMEWORK} MATCHES "CAFFE" OR ${DL_FRAMEWORK} MATCHES "NV_CAFFE")
  174. # Set the acceleration library
  175. if (APPLE)
  176. set(GPU_MODE CPU_ONLY CACHE STRING "Select the acceleration GPU library or CPU otherwise.")
  177. else (APPLE)
  178. set(GPU_MODE CUDA CACHE STRING "Select the acceleration GPU library or CPU otherwise.")
  179. endif (APPLE)
  180. # Display proper options to user
  181. # if (CUDA_FOUND AND OpenCL_FOUND)
  182. # set_property(CACHE GPU_MODE PROPERTY STRINGS CUDA OPENCL CPU_ONLY)
  183. # elseif (CUDA_FOUND)
  184. # set_property(CACHE GPU_MODE PROPERTY STRINGS CUDA CPU_ONLY)
  185. # elseif (OpenCL_FOUND)
  186. # set_property(CACHE GPU_MODE PROPERTY STRINGS OPENCL CPU_ONLY)
  187. # else ()
  188. # set_property(CACHE GPU_MODE PROPERTY STRINGS CPU_ONLY)
  189. # endif (CUDA_FOUND AND OpenCL_FOUND)
  190. set_property(CACHE GPU_MODE PROPERTY STRINGS CUDA OPENCL CPU_ONLY)
  191. # Look for CUDA
  192. set(CUDA_FOUND FALSE)
  193. if (${GPU_MODE} MATCHES "CUDA")
  194. find_package(CUDA)
  195. endif (${GPU_MODE} MATCHES "CUDA")
  196. # Look for OpenCL
  197. set(OpenCL_FOUND FALSE)
  198. set(CUDA_VERSION_MAJOR 0)
  199. if (${GPU_MODE} MATCHES "OPENCL")
  200. find_package(OpenCL)
  201. endif (${GPU_MODE} MATCHES "OPENCL")
  202. # Code to avoid crash at compiling time if OpenCL is not found
  203. if (NOT OpenCL_FOUND)
  204. set(OpenCL_LIBRARIES "")
  205. endif (NOT OpenCL_FOUND)
  206. # Required for OpenCL in Nvidia graphic cards
  207. if (CUDA_FOUND AND OpenCL_FOUND AND ${CUDA_VERSION_MAJOR} LESS 9)
  208. add_definitions(-DLOWER_CL_VERSION)
  209. endif (CUDA_FOUND AND OpenCL_FOUND AND ${CUDA_VERSION_MAJOR} LESS 9)
  210. # Handle desired GPU mode option
  211. if (${GPU_MODE} MATCHES "CUDA")
  212. # OpenPose flags
  213. add_definitions(-DUSE_CUDA)
  214. message(STATUS "Building with CUDA.")
  215. elseif (${GPU_MODE} MATCHES "CPU_ONLY")
  216. add_definitions(-DUSE_CPU_ONLY)
  217. message(STATUS "Building CPU Only.")
  218. # OpenPose flag for Caffe
  219. add_definitions(-DCPU_ONLY)
  220. elseif (${GPU_MODE} MATCHES "OPENCL")
  221. # OpenPose flag for Caffe
  222. add_definitions(-DUSE_OPENCL)
  223. add_definitions(-DUSE_GREENTEA)
  224. message(STATUS "Building with OpenCL.")
  225. endif (${GPU_MODE} MATCHES "CUDA")
  226. # Intel branch with MKL Support
  227. if (${GPU_MODE} MATCHES "CPU_ONLY")
  228. if (UNIX AND NOT APPLE)
  229. OPTION(USE_MKL "Use MKL Intel Branch." ON)
  230. endif (UNIX AND NOT APPLE)
  231. endif (${GPU_MODE} MATCHES "CPU_ONLY")
  232. if (${USE_MKL})
  233. # OpenPose flags
  234. add_definitions(-DUSE_MKL)
  235. message(STATUS "Building with MKL support.")
  236. endif (${USE_MKL})
  237. # Set/disable profiler
  238. if (PROFILER_ENABLED)
  239. add_definitions(-DPROFILER_ENABLED)
  240. endif (PROFILER_ENABLED)
  241. # Suboptions for GPU architectures
  242. if (${GPU_MODE} MATCHES "CUDA")
  243. set(CUDA_ARCH Auto CACHE STRING "Select target NVIDIA GPU achitecture.")
  244. set_property(CACHE CUDA_ARCH PROPERTY STRINGS Auto All Manual)
  245. endif (${GPU_MODE} MATCHES "CUDA")
  246. # Suboptions for acceleration library
  247. if (${GPU_MODE} MATCHES "CUDA")
  248. option(USE_CUDNN "Build OpenPose with cuDNN library support." ON)
  249. endif (${GPU_MODE} MATCHES "CUDA")
  250. # Suboptions for OpenPose 3D Reconstruction module and demo
  251. option(WITH_3D_RENDERER "Add OpenPose 3D renderer module (it requires FreeGLUT library)." OFF)
  252. if (UNIX AND NOT APPLE)
  253. option(WITH_CERES "Add Ceres support for advanced 3-D reconstruction." OFF)
  254. endif (UNIX AND NOT APPLE)
  255. option(WITH_FLIR_CAMERA "Add FLIR (formerly Point Grey) camera code (requires Spinnaker SDK already installed)." OFF)
  256. # option(WITH_3D_ADAM_MODEL "Add 3-D Adam model (requires OpenGL, Ceres, Eigen, OpenMP, FreeImage, GLEW, and IGL already installed)." OFF)
  257. # Faster GUI rendering
  258. # Note: It seems to work by default in Windows and Ubuntu, but not in Mac nor Android.
  259. # More info: https://stackoverflow.com/questions/21129683/does-opengl-display-image-faster-than-opencv?answertab=active#tab-top
  260. option(WITH_OPENCV_WITH_OPENGL "Much faster GUI display, but you must also enable OpenGL support in OpenCV by configuring OpenCV using CMake with WITH_OPENGL=ON flag." OFF)
  261. # Set the acceleration library
  262. if (WIN32 OR APPLE)
  263. set(WITH_EIGEN NONE CACHE STRING "Select the Eigen mode: NONE if not required, BUILD to let OpenPose download it, or FIND to let CMake find it (e.g., if you installed it manually).")
  264. set_property(CACHE WITH_EIGEN PROPERTY STRINGS NONE BUILD FIND)
  265. elseif (UNIX AND NOT APPLE)
  266. set(WITH_EIGEN NONE CACHE STRING "Select the Eigen mode: NONE if not required, BUILD to let OpenPose download it, or FIND to let CMake find it (e.g., if you installed it manually or used `sudo apt-get install libeigen3-dev`).")
  267. set_property(CACHE WITH_EIGEN PROPERTY STRINGS NONE BUILD FIND)
  268. endif (WIN32 OR APPLE)
  269. # # Suboptions for OpenPose tracking
  270. # if (UNIX AND NOT APPLE)
  271. # option(WITH_TRACKING "Add OpenPose 3D tracking module (it requires OpenCV with CUDA support)." OFF)
  272. # endif (UNIX AND NOT APPLE)
  273. # Download the models
  274. option(DOWNLOAD_BODY_25_MODEL "Download body 25-keypoint (body COCO and 6-keypoint foot) model." ON)
  275. option(DOWNLOAD_BODY_COCO_MODEL "Download body 18-keypoint COCO model." OFF)
  276. option(DOWNLOAD_BODY_MPI_MODEL "Download body 15-keypoint MPI model." OFF)
  277. option(DOWNLOAD_FACE_MODEL "Download face model." ON)
  278. option(DOWNLOAD_HAND_MODEL "Download hand model." ON)
  279. # Asio
  280. # option(USE_ASIO "Include Asio header-only library." OFF)
  281. # More options
  282. option(BUILD_EXAMPLES "Build OpenPose examples." ON)
  283. option(BUILD_DOCS "Build OpenPose documentation." OFF)
  284. option(BUILD_PYTHON "Build OpenPose python." OFF)
  285. if (WIN32)
  286. option(BUILD_BIN_FOLDER "Copy all required 3rd-party DLL files into {build_directory}/bin. Disable to save some memory." ON)
  287. endif (WIN32)
  288. # Unity
  289. option(BUILD_UNITY_SUPPORT "Build OpenPose as a Unity plugin." OFF)
  290. # Build as shared library
  291. option(BUILD_SHARED_LIBS "Build as shared lib." ON)
  292. # Speed profiler
  293. option(PROFILER_ENABLED "If enabled, OpenPose will be able to print out speed information at runtime." OFF)
  294. # Threads - Pthread
  295. if (${GPU_MODE} MATCHES "OPENCL" OR (UNIX OR APPLE))
  296. unset(CMAKE_THREAD_LIBS_INIT CACHE)
  297. find_package(Threads)
  298. endif (${GPU_MODE} MATCHES "OPENCL" OR (UNIX OR APPLE))
  299. # Caffe OpenCL Boost Issue
  300. if (APPLE)
  301. if (${GPU_MODE} MATCHES "OPENCL" OR BUILD_PYTHON)
  302. unset(Boost_SYSTEM_LIBRARY CACHE)
  303. find_package(Boost COMPONENTS system REQUIRED)
  304. else ()
  305. set(Boost_SYSTEM_LIBRARY "")
  306. endif ()
  307. endif (APPLE)
  308. ### FIND REQUIRED PACKAGES
  309. list(APPEND CMAKE_MODULE_PATH "${CMAKE_CURRENT_SOURCE_DIR}/cmake/Modules")
  310. if (${GPU_MODE} MATCHES "CUDA")
  311. find_package(CUDA)
  312. endif (${GPU_MODE} MATCHES "CUDA")
  313. # Adding 3D
  314. if (WITH_OPENCV_WITH_OPENGL)
  315. # OpenPose flags
  316. add_definitions(-DUSE_OPENCV_WITH_OPENGL)
  317. endif (WITH_OPENCV_WITH_OPENGL)
  318. if (WITH_3D_RENDERER)
  319. # OpenPose flags
  320. add_definitions(-DUSE_3D_RENDERER)
  321. endif (WITH_3D_RENDERER)
  322. if (WITH_CERES)
  323. add_definitions(-DUSE_CERES)
  324. endif (WITH_CERES)
  325. if (WITH_FLIR_CAMERA)
  326. # OpenPose flags
  327. add_definitions(-DUSE_FLIR_CAMERA)
  328. endif (WITH_FLIR_CAMERA)
  329. if (WITH_3D_ADAM_MODEL)
  330. # OpenPose flags
  331. add_definitions(-DUSE_3D_ADAM_MODEL)
  332. endif (WITH_3D_ADAM_MODEL)
  333. # Adding tracking
  334. if (WITH_TRACKING)
  335. # OpenPose flags
  336. add_definitions(-DUSE_TRACKING)
  337. endif (WITH_TRACKING)
  338. # Unity
  339. if (BUILD_UNITY_SUPPORT)
  340. # OpenPose flags
  341. add_definitions(-DUSE_UNITY_SUPPORT)
  342. endif (BUILD_UNITY_SUPPORT)
  343. # Calibration
  344. # No Eigen
  345. if (${WITH_EIGEN} MATCHES "NONE")
  346. if (WITH_CERES)
  347. message(FATAL_ERROR "Eigen is required (enable `WITH_EIGEN`) if CERES is enabled.")
  348. endif (WITH_CERES)
  349. # If Eigen used
  350. else (${WITH_EIGEN} MATCHES "NONE")
  351. # OpenPose flags
  352. add_definitions(-DUSE_EIGEN)
  353. # OpenPose download/builds Eigen
  354. if (${WITH_EIGEN} MATCHES "BUILD")
  355. # Download it
  356. set(OP_URL "http://posefs1.perception.cs.cmu.edu/OpenPose/3rdparty/")
  357. set(FIND_LIB_PREFIX ${CMAKE_SOURCE_DIR}/3rdparty/)
  358. download_zip("eigen_2018_05_23.zip" ${OP_URL} ${FIND_LIB_PREFIX} 29B9B2FD4679D587BB67467F09EE8365)
  359. # Set path
  360. set(EIGEN3_INCLUDE_DIRS "3rdparty/eigen/include/")
  361. # Leave to CMake to find it
  362. elseif (${WITH_EIGEN} MATCHES "FIND")
  363. # Eigen
  364. # # Option a) apt-get
  365. # find_package(PkgConfig)
  366. # pkg_check_modules(EIGEN3 REQUIRED eigen3)
  367. # # Option b) Find it (including apt-get)
  368. if (CUDA_VERSION_STRING STRGREATER "9.0")
  369. find_package(Eigen3 3.3.5 REQUIRED NO_MODULE)
  370. else (CUDA_VERSION_STRING STRGREATER "9.0")
  371. find_package(Eigen3 3 REQUIRED NO_MODULE)
  372. endif (CUDA_VERSION_STRING STRGREATER "9.0")
  373. endif (${WITH_EIGEN} MATCHES "BUILD")
  374. endif (${WITH_EIGEN} MATCHES "NONE")
  375. if (UNIX OR APPLE)
  376. if (${GPU_MODE} MATCHES "CUDA")
  377. include(cmake/Cuda.cmake)
  378. find_package(CuDNN)
  379. endif (${GPU_MODE} MATCHES "CUDA")
  380. find_package(GFlags) # For Caffe and OpenPose
  381. find_package(Glog) # For Caffe
  382. find_package(Protobuf REQUIRED) # For Caffe
  383. if (OpenCV_CONFIG_FILE)
  384. include (${OpenCV_CONFIG_FILE})
  385. # Allow explicitly setting the OpenCV includes and libs
  386. elseif (OpenCV_INCLUDE_DIRS AND OpenCV_LIBS)
  387. set(OpenCV_FOUND 1)
  388. elseif (OpenCV_INCLUDE_DIRS AND OpenCV_LIBS_DIR)
  389. file(GLOB_RECURSE OpenCV_LIBS "${OpenCV_LIBS_DIR}*.so")
  390. set(OpenCV_FOUND 1)
  391. else (OpenCV_CONFIG_FILE)
  392. find_package(OpenCV)
  393. endif (OpenCV_CONFIG_FILE)
  394. # 3D
  395. if (WITH_3D_RENDERER)
  396. # OpenGL
  397. find_package(OpenGL REQUIRED)
  398. # GLUT
  399. find_package(GLUT REQUIRED)
  400. # OpenGL
  401. find_package(OpenGL REQUIRED)
  402. endif (WITH_3D_RENDERER)
  403. if (WITH_CERES)
  404. # Eigen + Ceres
  405. find_package(Ceres REQUIRED COMPONENTS SuiteSparse)
  406. endif (WITH_CERES)
  407. if (WITH_FLIR_CAMERA)
  408. # Spinnaker
  409. find_package(Spinnaker)
  410. if (NOT SPINNAKER_FOUND)
  411. message(FATAL_ERROR "Spinnaker not found. Either turn off the `WITH_FLIR_CAMERA` option or specify the path to
  412. the Spinnaker includes and libs.")
  413. endif (NOT SPINNAKER_FOUND)
  414. endif (WITH_FLIR_CAMERA)
  415. if (WITH_3D_ADAM_MODEL)
  416. if (NOT WITH_3D_RENDERER)
  417. message(FATAL_ERROR "WITH_3D_RENDERER is required if WITH_3D_ADAM_MODEL is enabled.")
  418. endif (NOT WITH_3D_RENDERER)
  419. # Eigen
  420. # # Option a) apt-get
  421. # find_package(PkgConfig)
  422. # pkg_check_modules(EIGEN3 REQUIRED eigen3)
  423. # # Option b) Find it (including apt-get)
  424. if (CUDA_VERSION_STRING STRGREATER "9.0")
  425. find_package(Eigen3 3.3.5 REQUIRED NO_MODULE)
  426. else (CUDA_VERSION_STRING STRGREATER "9.0")
  427. find_package(Eigen3 3 REQUIRED NO_MODULE)
  428. endif (CUDA_VERSION_STRING STRGREATER "9.0")
  429. # Others: sudo apt-get install libglm-dev
  430. # http://ceres-solver.org
  431. find_package(Ceres REQUIRED COMPONENTS SuiteSparse)
  432. # sudo apt-get install libglew-dev
  433. find_package(GLEW REQUIRED)
  434. # find_package(GLUT REQUIRED) # TODO: Duplicated of WITH_3D_RENDERER, clean somehow (like Eigen)
  435. # git clone --recursive https://github.com/libigl/libigl.git
  436. # No installation, it's header only
  437. # TODO: It's header only (as Eigen), do BUILD option too to download it
  438. find_package(LIBIGL REQUIRED)
  439. find_package(OpenMP REQUIRED)
  440. # Only adam/renderer::Renderer::IdleSaveImage() uses it. Make dependency optional in hand_model
  441. # FIND_LIBRARY(FREE_IMAGE_LIBRARY NAMES libfreeimage.so)
  442. set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${OpenMP_C_FLAGS}")
  443. set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${OpenMP_CXX_FLAGS}")
  444. set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} ${OpenMP_EXE_LINKER_FLAGS}")
  445. endif (WITH_3D_ADAM_MODEL)
  446. # OpenMP
  447. if (${GPU_MODE} MATCHES "CPU_ONLY")
  448. find_package(OpenMP)
  449. if (OPENMP_FOUND)
  450. set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${OpenMP_C_FLAGS}")
  451. set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${OpenMP_CXX_FLAGS}")
  452. set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} ${OpenMP_EXE_LINKER_FLAGS}")
  453. endif (OPENMP_FOUND)
  454. endif (${GPU_MODE} MATCHES "CPU_ONLY")
  455. if (${GPU_MODE} MATCHES "CUDA")
  456. # Set CUDA Flags
  457. set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} -std=c++11")
  458. if (NOT CUDA_FOUND)
  459. message(STATUS "CUDA not found.")
  460. execute_process(COMMAND cat install_cuda.sh WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/scripts/ubuntu)
  461. message(FATAL_ERROR "Install CUDA using the above commands.")
  462. endif (NOT CUDA_FOUND)
  463. if (USE_CUDNN AND NOT CUDNN_FOUND)
  464. message(STATUS "cuDNN not found.")
  465. execute_process(COMMAND cat install_cudnn.sh WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/scripts/ubuntu)
  466. message(FATAL_ERROR "Install cuDNN using the above commands. or turn off cuDNN by setting USE_CUDNN to OFF.")
  467. endif (USE_CUDNN AND NOT CUDNN_FOUND)
  468. endif (${GPU_MODE} MATCHES "CUDA")
  469. if (NOT GLOG_FOUND)
  470. message(FATAL_ERROR "Glog not found. Install Glog from the command line using the command(s) -\
  471. sudo apt-get install libgoogle-glog-dev")
  472. endif (NOT GLOG_FOUND)
  473. if (NOT GFLAGS_FOUND)
  474. message(FATAL_ERROR "GFlags not found. Install GFlags from the command line using the command(s) --\
  475. sudo apt-get install libgflags-dev")
  476. endif (NOT GFLAGS_FOUND)
  477. if (NOT OpenCV_FOUND)
  478. message(FATAL_ERROR "OpenCV not found. Install OpenCV from the command line using the command(s) --\
  479. sudo apt-get install libopencv-dev")
  480. endif (NOT OpenCV_FOUND)
  481. endif (UNIX OR APPLE)
  482. if (WIN32)
  483. if ("${CMAKE_VERSION}" VERSION_GREATER 3.6.3)
  484. set_property(DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR} PROPERTY VS_STARTUP_PROJECT OpenPoseDemo)
  485. endif ("${CMAKE_VERSION}" VERSION_GREATER 3.6.3)
  486. set_property(GLOBAL PROPERTY USE_FOLDERS ON)
  487. set(FIND_LIB_PREFIX ${CMAKE_SOURCE_DIR}/3rdparty/windows)
  488. # Download Windows 3rd party
  489. message(STATUS "Downloading windows dependencies...")
  490. set(OP_WIN_URL "http://posefs1.perception.cs.cmu.edu/OpenPose/3rdparty/windows")
  491. set(OP_WIN_DIR "${CMAKE_SOURCE_DIR}/3rdparty/windows")
  492. # Download required zip files
  493. download_zip("opencv_411_v14_15_2019_09_24.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} cf080121a8189ecb7f53ac9b0fef36c6)
  494. # download_zip("caffe3rdparty_2019_03_12.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 34c49be6a2418fd1670b46fc455eebc0)
  495. download_zip("caffe3rdparty_15_2019_03_14.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 9fcc6bdf4765db699d1a279c8d6b830b)
  496. if (${GPU_MODE} MATCHES "OPENCL")
  497. download_zip("caffe_opencl_2018_02_13.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 3ac3e1acf5ee6a4e57920be73053067a)
  498. elseif (${GPU_MODE} MATCHES "CPU_ONLY")
  499. download_zip("caffe_cpu_2018_05_27.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 87E8401B6DFBAC5B8E909DD20E3B3390)
  500. else (${GPU_MODE} MATCHES "OPENCL")
  501. # download_zip("caffe_2019_03_12.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 859a592310f0928fd4f40da1456a217f)
  502. download_zip("caffe_15_2019_05_16.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} 12F7A675A3276CD33C76B37D21C2F421)
  503. endif (${GPU_MODE} MATCHES "OPENCL")
  504. if (WITH_3D_RENDERER)
  505. download_zip("freeglut_2018_01_14.zip" ${OP_WIN_URL} ${FIND_LIB_PREFIX} BB182187285E06880F0EDE3A39530091)
  506. endif (WITH_3D_RENDERER)
  507. message(STATUS "Windows dependencies downloaded.")
  508. # set(VS_VERSION "14")
  509. set(VS_VERSION "15") # Note: VS2017 code seems to simply works in VS2015
  510. find_library(OpenCV_LIBS opencv_world411 HINTS ${FIND_LIB_PREFIX}/opencv/x64/vc${VS_VERSION}/lib)
  511. find_library(GFLAGS_LIBRARY_RELEASE gflags HINTS ${FIND_LIB_PREFIX}/caffe3rdparty/lib)
  512. find_library(GFLAGS_LIBRARY_DEBUG gflagsd HINTS ${FIND_LIB_PREFIX}/caffe3rdparty/lib)
  513. find_library(GLOG_LIBRARY_RELEASE glog HINTS ${FIND_LIB_PREFIX}/caffe3rdparty/lib)
  514. find_library(GLOG_LIBRARY_DEBUG glogd HINTS ${FIND_LIB_PREFIX}/caffe3rdparty/lib)
  515. # If OpenPose builds it
  516. if (BUILD_CAFFE)
  517. unset(Caffe_INCLUDE_DIRS CACHE)
  518. unset(Caffe_LIB CACHE)
  519. unset(Caffe_Proto_LIB CACHE)
  520. endif (BUILD_CAFFE)
  521. # OpenCL
  522. if (${GPU_MODE} MATCHES "OPENCL")
  523. set(VCXPROJ_FILE_GPU_MODE "_CL")
  524. find_library(Caffe_LIB caffe HINTS ${FIND_LIB_PREFIX}/caffe_opencl/lib)
  525. find_library(Caffe_Proto_LIB caffeproto HINTS ${FIND_LIB_PREFIX}/caffe_opencl/lib)
  526. # CPU & CUDA
  527. else (${GPU_MODE} MATCHES "OPENCL")
  528. # CPU
  529. if (${GPU_MODE} MATCHES "CPU_ONLY")
  530. set(VCXPROJ_FILE_GPU_MODE "_CPU")
  531. find_library(Caffe_LIB caffe HINTS ${FIND_LIB_PREFIX}/caffe_cpu/lib)
  532. find_library(Caffe_Proto_LIB caffeproto HINTS ${FIND_LIB_PREFIX}/caffe_cpu/lib)
  533. # CUDA
  534. else (${GPU_MODE} MATCHES "CPU_ONLY")
  535. set(VCXPROJ_FILE_GPU_MODE "")
  536. find_library(Caffe_LIB caffe HINTS ${FIND_LIB_PREFIX}/caffe/lib)
  537. find_library(Caffe_Proto_LIB caffeproto HINTS ${FIND_LIB_PREFIX}/caffe/lib)
  538. endif (${GPU_MODE} MATCHES "CPU_ONLY")
  539. endif (${GPU_MODE} MATCHES "OPENCL")
  540. # Boost DepCopy over required DLL F
  541. if (${GPU_MODE} MATCHES "CPU_ONLY" OR ${GPU_MODE} MATCHES "OPENCL" OR BUILD_PYTHON)
  542. find_library(BOOST_SYSTEM_LIB_RELEASE boost_system-vc141-mt-x64-1_69 HINTS ${FIND_LIB_PREFIX}/caffe3rdparty/lib)
  543. find_library(BOOST_SYSTEM_LIB_DEBUG boost_system-vc141-mt-gd-x64-1_69 HINTS ${FIND_LIB_PREFIX}/caffe3rdparty/lib)
  544. find_library(BOOST_FILESYSTEM_LIB_RELEASE boost_filesystem-vc141-mt-x64-1_69 HINTS ${FIND_LIB_PREFIX}/caffe3rdparty/lib)
  545. find_library(BOOST_FILESYSTEM_LIB_DEBUG boost_filesystem-vc141-mt-gd-x64-1_69 HINTS ${FIND_LIB_PREFIX}/caffe3rdparty/lib)
  546. else ()
  547. set(BOOST_SYSTEM_LIB_RELEASE "")
  548. set(BOOST_SYSTEM_LIB_DEBUG "")
  549. set(BOOST_FILESYSTEM_LIB_RELEASE "")
  550. set(BOOST_FILESYSTEM_LIB_DEBUG "")
  551. endif ()
  552. if (WITH_3D_RENDERER)
  553. find_library(GLUT_LIBRARY freeglut HINTS ${FIND_LIB_PREFIX}/freeglut/lib)
  554. message(STATUS "\${GLUT_LIBRARY} = ${GLUT_LIBRARY}")
  555. endif (WITH_3D_RENDERER)
  556. if (WITH_FLIR_CAMERA)
  557. find_library(SPINNAKER_LIB spinnaker_v140 HINTS ${FIND_LIB_PREFIX}/spinnaker/lib)
  558. endif (WITH_FLIR_CAMERA)
  559. set(Caffe_LIBS ${Caffe_LIB};${Caffe_Proto_LIB})
  560. set(OpenCV_INCLUDE_DIRS "3rdparty/windows/opencv/include")
  561. # OpenCL
  562. if (${GPU_MODE} MATCHES "OPENCL")
  563. unset(Caffe_INCLUDE_DIRS CACHE)
  564. set(Caffe_INCLUDE_DIRS "3rdparty/windows/caffe_opencl/include;3rdparty/windows/caffe_opencl/include2" CACHE FILEPATH "Caffe_INCLUDE_DIRS")
  565. # CUDA and CPU
  566. else (${GPU_MODE} MATCHES "OPENCL")
  567. # CPU
  568. if (${GPU_MODE} MATCHES "CPU_ONLY")
  569. set(Caffe_INCLUDE_DIRS "3rdparty/windows/caffe_cpu/include;3rdparty/windows/caffe_cpu/include2" CACHE FILEPATH "Caffe_INCLUDE_DIRS")
  570. # CUDA
  571. else (${GPU_MODE} MATCHES "CPU_ONLY")
  572. set(Caffe_INCLUDE_DIRS "3rdparty/windows/caffe/include;3rdparty/windows/caffe/include2" CACHE FILEPATH "Caffe_INCLUDE_DIRS")
  573. endif (${GPU_MODE} MATCHES "CPU_ONLY")
  574. endif (${GPU_MODE} MATCHES "OPENCL")
  575. # set(Boost_INCLUDE_DIRS "3rdparty/windows/caffe3rdparty/include/boost-1_61") # Added in WINDOWS_INCLUDE_DIRS
  576. set(WINDOWS_INCLUDE_DIRS "3rdparty/windows/caffe3rdparty/include")
  577. if (WITH_3D_RENDERER)
  578. set(GLUT_INCLUDE_DIRS "3rdparty/windows/freeglut/include")
  579. endif (WITH_3D_RENDERER)
  580. if (WITH_FLIR_CAMERA)
  581. set(SPINNAKER_INCLUDE_DIRS "3rdparty/windows/spinnaker/include")
  582. endif (WITH_FLIR_CAMERA)
  583. set(Caffe_FOUND 1)
  584. # Build DLL Must be on if Build Python is on
  585. if (BUILD_PYTHON)
  586. if (NOT BUILD_BIN_FOLDER)
  587. message(FATAL_ERROR "BUILD_BIN_FOLDER must be turned on to as well to build python library")
  588. endif (NOT BUILD_BIN_FOLDER)
  589. endif (BUILD_PYTHON)
  590. # Auto copy DLLs
  591. if (BUILD_BIN_FOLDER)
  592. # Locate DLLs
  593. # Caffe DLLs
  594. if (${GPU_MODE} MATCHES "CUDA")
  595. file(GLOB CAFFE_DLL "${CMAKE_SOURCE_DIR}/3rdparty/windows/caffe/bin/*.dll")
  596. elseif (${GPU_MODE} MATCHES "OPENCL")
  597. file(GLOB CAFFE_DLL "${CMAKE_SOURCE_DIR}/3rdparty/windows/caffe_opencl/bin/*.dll")
  598. elseif (${GPU_MODE} MATCHES "CPU_ONLY")
  599. file(GLOB CAFFE_DLL "${CMAKE_SOURCE_DIR}/3rdparty/windows/caffe_cpu/bin/*.dll")
  600. endif ()
  601. # Caffe 3rd-party DLLs
  602. file(GLOB CAFFE_3RD_PARTY_DLL "${CMAKE_SOURCE_DIR}/3rdparty/windows/caffe3rdparty/lib/*.dll")
  603. # OpenCV DLLs
  604. file(GLOB OPENCV_DLL "${CMAKE_SOURCE_DIR}/3rdparty/windows/opencv/x64/vc${VS_VERSION}/bin/*.dll")
  605. # GLUT DLLs
  606. file(GLOB GLUT_DLL "${CMAKE_SOURCE_DIR}/3rdparty/windows/freeglut/bin/*.dll")
  607. # Spinnaker DLLs and other files
  608. file(GLOB SPINNAKER_DLL "${CMAKE_SOURCE_DIR}/3rdparty/windows/spinnaker/bin/*")
  609. # Copy DLLs into same folder
  610. set(BIN_FOLDER ${CMAKE_BINARY_DIR}/bin)
  611. file(MAKE_DIRECTORY ${BIN_FOLDER})
  612. file(COPY ${CAFFE_DLL} DESTINATION ${BIN_FOLDER})
  613. file(COPY ${OPENCV_DLL} DESTINATION ${BIN_FOLDER})
  614. file(COPY ${CAFFE_3RD_PARTY_DLL} DESTINATION ${BIN_FOLDER})
  615. file(COPY ${GLUT_DLL} DESTINATION ${BIN_FOLDER})
  616. file(COPY ${SPINNAKER_DLL} DESTINATION ${BIN_FOLDER})
  617. endif (BUILD_BIN_FOLDER)
  618. endif (WIN32)
  619. ### CAFFE/NVCAFFE in Ubuntu
  620. if (UNIX OR APPLE)
  621. if (${DL_FRAMEWORK} MATCHES "CAFFE" OR ${DL_FRAMEWORK} MATCHES "NV_CAFFE")
  622. # Check if the user specified caffe paths
  623. if (Caffe_INCLUDE_DIRS AND Caffe_LIBS AND NOT BUILD_CAFFE)
  624. message(STATUS "\${Caffe_INCLUDE_DIRS} set by the user to " ${Caffe_INCLUDE_DIRS})
  625. message(STATUS "\${Caffe_LIBS} set by the user to " ${Caffe_LIBS})
  626. set(Caffe_FOUND 1)
  627. endif (Caffe_INCLUDE_DIRS AND Caffe_LIBS AND NOT BUILD_CAFFE)
  628. # Else build from scratch
  629. if (BUILD_CAFFE)
  630. # Download Caffe
  631. message(STATUS "Caffe will be downloaded from source now. NOTE: This process might take several minutes depending
  632. on your internet connection.")
  633. # Check if pulled
  634. file(GLOB CAFFE_DIR_VALID ${CMAKE_SOURCE_DIR}/3rdparty/caffe/*)
  635. list(LENGTH CAFFE_DIR_VALID CAFFE_DIR_VALID_LENGTH)
  636. if (CAFFE_DIR_VALID_LENGTH EQUAL 0)
  637. execute_process(COMMAND git submodule update --init ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
  638. # execute_process(COMMAND git submodule update --init --recursive --remote) # This would initialize them all
  639. else (CAFFE_DIR_VALID_LENGTH EQUAL 0)
  640. message(STATUS "Caffe has already been downloaded.")
  641. endif (CAFFE_DIR_VALID_LENGTH EQUAL 0)
  642. # Build Process
  643. set(CAFFE_CPU_ONLY OFF)
  644. if (${GPU_MODE} MATCHES "CUDA")
  645. # execute_process(COMMAND git checkout master WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
  646. execute_process(COMMAND git checkout c95002f WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
  647. elseif (${GPU_MODE} MATCHES "CPU_ONLY")
  648. if (USE_MKL)
  649. #execute_process(COMMAND git checkout intel WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
  650. execute_process(COMMAND git checkout b6712ce WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
  651. execute_process(COMMAND sh prepare_mkl.sh WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe/external/mkl
  652. OUTPUT_VARIABLE rv)
  653. set( MLIST ${rv} )
  654. separate_arguments(MLIST)
  655. list(GET MLIST 0 MKL_PATH)
  656. message(STATUS ${MKL_PATH})
  657. file(GLOB MKL_SO
  658. "${MKL_PATH}lib/*"
  659. )
  660. file(COPY ${MKL_SO} DESTINATION ${CMAKE_BINARY_DIR}/caffe)
  661. # New MLSL Lib
  662. #execute_process(COMMAND sh prepare_mlsl.sh WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe/external/mlsl
  663. # OUTPUT_VARIABLE rv)
  664. #set( MLIST ${rv} )
  665. #separate_arguments(MLIST)
  666. #list(GET MLIST 0 MLSL_PATH)
  667. #message(STATUS ${MLSL_PATH})
  668. #file(GLOB MLSL_SO
  669. # "${MLSL_PATH}/intel64/lib/*"
  670. # )
  671. #file(COPY ${MLSL_SO} DESTINATION ${CMAKE_BINARY_DIR}/caffe)
  672. set(MKL_LIBS
  673. #"${CMAKE_BINARY_DIR}/caffe/libmlsl.so"
  674. "${CMAKE_BINARY_DIR}/caffe/libiomp5.so"
  675. "${CMAKE_BINARY_DIR}/caffe/libmklml_intel.so"
  676. "${CMAKE_BINARY_DIR}/caffe/lib/libmkldnn.so"
  677. )
  678. else (USE_MKL)
  679. # execute_process(COMMAND git checkout master WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
  680. execute_process(COMMAND git checkout c95002f WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
  681. endif (USE_MKL)
  682. set(CAFFE_CPU_ONLY ON)
  683. set(USE_CUDNN OFF)
  684. elseif (${GPU_MODE} MATCHES "OPENCL")
  685. execute_process(COMMAND git checkout fe2a1102 WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
  686. set(USE_CUDNN OFF)
  687. endif (${GPU_MODE} MATCHES "CUDA")
  688. # Build Caffe
  689. message(STATUS "Caffe will be built from source now.")
  690. find_package(Caffe)
  691. include(ExternalProject)
  692. set(CAFFE_PREFIX caffe)
  693. set(CAFFE_URL ${CMAKE_SOURCE_DIR}/3rdparty/caffe)
  694. # One for Intel Branch and one for Master
  695. if (USE_MKL)
  696. ExternalProject_Add(openpose_lib
  697. SOURCE_DIR ${CAFFE_URL}
  698. PREFIX ${CAFFE_PREFIX}
  699. CMAKE_ARGS -DCMAKE_INSTALL_PREFIX:PATH=<INSTALL_DIR>
  700. -DCMAKE_TOOLCHAIN_FILE=${CMAKE_TOOLCHAIN_FILE}
  701. -DMKLDNN_INSTALL_DIR:PATH=<INSTALL_DIR>
  702. -DUSE_MKL2017_AS_DEFAULT_ENGINE=${CAFFE_CPU_ONLY}
  703. -DUSE_CUDNN=${USE_CUDNN}
  704. -DCUDA_ARCH_NAME=${CUDA_ARCH}
  705. -DCUDA_ARCH_BIN=${CUDA_ARCH_BIN}
  706. -DCUDA_ARCH_PTX=${CUDA_ARCH_PTX}
  707. -DCPU_ONLY=${CAFFE_CPU_ONLY}
  708. -DCMAKE_BUILD_TYPE=Release
  709. -DBUILD_docs=OFF
  710. -DBUILD_python=OFF
  711. -DBUILD_python_layer=OFF
  712. -DUSE_LEVELDB=OFF
  713. -DUSE_LMDB=OFF
  714. -DUSE_OPENCV=OFF)
  715. # -DOpenCV_DIR=${OpenCV_DIR})
  716. else (USE_MKL)
  717. ExternalProject_Add(openpose_lib
  718. SOURCE_DIR ${CAFFE_URL}
  719. PREFIX ${CAFFE_PREFIX}
  720. CMAKE_ARGS -DCMAKE_INSTALL_PREFIX:PATH=<INSTALL_DIR>
  721. -DCMAKE_TOOLCHAIN_FILE=${CMAKE_TOOLCHAIN_FILE}
  722. -DUSE_CUDNN=${USE_CUDNN}
  723. -DCUDA_ARCH_NAME=${CUDA_ARCH}
  724. -DCUDA_ARCH_BIN=${CUDA_ARCH_BIN}
  725. -DCUDA_ARCH_PTX=${CUDA_ARCH_PTX}
  726. -DCPU_ONLY=${CAFFE_CPU_ONLY}
  727. -DCMAKE_BUILD_TYPE=Release
  728. -DBUILD_docs=OFF
  729. -DBUILD_python=OFF
  730. -DBUILD_python_layer=OFF
  731. -DUSE_LEVELDB=OFF
  732. -DUSE_LMDB=OFF
  733. -DUSE_OPENCV=OFF)
  734. # -DOpenCV_DIR=${OpenCV_DIR})
  735. endif (USE_MKL)
  736. ExternalProject_Get_Property(openpose_lib install_dir)
  737. if (NOT Caffe_FOUND)
  738. add_custom_command(TARGET openpose_lib
  739. POST_BUILD
  740. COMMAND ${CMAKE_COMMAND} ${CMAKE_SOURCE_DIR}
  741. COMMAND $(MAKE)
  742. WORKING_DIRECTORY ${CMAKE_BINARY_DIR}
  743. COMMENT "Rerunning cmake after building Caffe submodule")
  744. endif (NOT Caffe_FOUND)
  745. endif (BUILD_CAFFE)
  746. if (NOT Caffe_FOUND AND NOT BUILD_CAFFE)
  747. message(FATAL_ERROR "Caffe not found. Either turn on the BUILD_CAFFE option or specify the path of Caffe includes
  748. and libs using -DCaffe_INCLUDE_DIRS and -DCaffe_LIBS.")
  749. endif (NOT Caffe_FOUND AND NOT BUILD_CAFFE)
  750. endif (${DL_FRAMEWORK} MATCHES "CAFFE" OR ${DL_FRAMEWORK} MATCHES "NV_CAFFE")
  751. endif (UNIX OR APPLE)
  752. ### PROJECT INCLUDES
  753. # Specify the include directories
  754. include_directories(
  755. include
  756. ${Protobuf_INCLUDE_DIRS}
  757. ${GFLAGS_INCLUDE_DIR}
  758. ${GLOG_INCLUDE_DIR}
  759. ${OpenCV_INCLUDE_DIRS})
  760. if (USE_ASIO)
  761. include_directories(${CMAKE_SOURCE_DIR}/3rdparty/asio/include/)
  762. # OpenPose flags
  763. add_definitions(-DUSE_ASIO)
  764. # Tell Asio it is not using Boost
  765. add_definitions(-DASIO_STANDALONE)
  766. endif (USE_ASIO)
  767. # Calibration
  768. if (NOT ${WITH_EIGEN} MATCHES "NONE")
  769. include_directories(
  770. ${EIGEN3_INCLUDE_DIRS})
  771. set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries} Eigen3::Eigen)
  772. endif (NOT ${WITH_EIGEN} MATCHES "NONE")
  773. if (APPLE)
  774. include_directories(
  775. "/usr/local/opt/openblas/include")
  776. endif (APPLE)
  777. if (USE_MKL)
  778. include_directories(
  779. "${MKL_PATH}/include/")
  780. endif (USE_MKL)
  781. if (Caffe_FOUND)
  782. include_directories(
  783. ${Caffe_INCLUDE_DIRS})
  784. endif (Caffe_FOUND)
  785. if (${GPU_MODE} MATCHES "CUDA")
  786. include_directories(
  787. ${CUDA_INCLUDE_DIRS})
  788. elseif (${GPU_MODE} MATCHES "OPENCL")
  789. include_directories(
  790. ${OpenCL_INCLUDE_DIRS})
  791. endif (${GPU_MODE} MATCHES "CUDA")
  792. # 3D
  793. if (WITH_3D_RENDERER)
  794. include_directories(${GLUT_INCLUDE_DIRS})
  795. endif (WITH_3D_RENDERER)
  796. if (WITH_CERES)
  797. include_directories(${CERES_INCLUDE_DIRS})
  798. endif (WITH_CERES)
  799. if (WITH_FLIR_CAMERA)
  800. include_directories(SYSTEM ${SPINNAKER_INCLUDE_DIRS}) # To remove its warnings, equiv. to -isystem
  801. endif (WITH_FLIR_CAMERA)
  802. if (WITH_3D_ADAM_MODEL)
  803. include_directories(include/adam) # TODO: TEMPORARY - TO BE REMOVED IN THE FUTURE
  804. include_directories(${CERES_INCLUDE_DIRS})
  805. include_directories(${EIGEN3_INCLUDE_DIRS})
  806. set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries} Eigen3::Eigen)
  807. include_directories(${IGL_INCLUDE_DIRS})
  808. include_directories(${LIBIGL_INCLUDE_DIRS})
  809. include_directories(${OPENGL_INCLUDE_DIR} ${GLUT_INCLUDE_DIRS} ${GLEW_INCLUDE_DIRS})
  810. endif (WITH_3D_ADAM_MODEL)
  811. # Windows includes
  812. if (WIN32)
  813. include_directories(
  814. # ${Boost_INCLUDE_DIRS}
  815. ${WINDOWS_INCLUDE_DIRS})
  816. endif (WIN32)
  817. ### COLLECT ALL 3RD-PARTY LIBRARIES TO BE LINKED AGAINST
  818. set(OpenPose_3rdparty_libraries ${OpenCV_LIBS} ${GLOG_LIBRARY})
  819. if (UNIX OR APPLE)
  820. set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries} ${GLOG_LIBRARY})
  821. elseif (WIN32)
  822. set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries}
  823. debug ${GFLAGS_LIBRARY_DEBUG} optimized ${GFLAGS_LIBRARY_RELEASE}
  824. debug ${GLOG_LIBRARY_DEBUG} optimized ${GLOG_LIBRARY_RELEASE})
  825. endif (UNIX OR APPLE)
  826. # Deep net Framework
  827. if (${DL_FRAMEWORK} MATCHES "CAFFE" OR ${DL_FRAMEWORK} MATCHES "NV_CAFFE")
  828. set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries} ${Caffe_LIBS} ${GFLAGS_LIBRARY})
  829. endif (${DL_FRAMEWORK} MATCHES "CAFFE" OR ${DL_FRAMEWORK} MATCHES "NV_CAFFE")
  830. # CPU vs. GPU
  831. if (USE_MKL)
  832. set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries} ${MKL_LIBS})
  833. endif (USE_MKL)
  834. if (${GPU_MODE} MATCHES "OPENCL")
  835. set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries} ${CMAKE_THREAD_LIBS_INIT} ${OpenCL_LIBRARIES})
  836. endif (${GPU_MODE} MATCHES "OPENCL")
  837. # Boost
  838. if (WIN32)
  839. if (${GPU_MODE} MATCHES "CPU_ONLY" OR ${GPU_MODE} MATCHES "OPENCL" OR BUILD_PYTHON)
  840. set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries}
  841. debug ${BOOST_SYSTEM_LIB_RELEASE} optimized ${BOOST_SYSTEM_LIB_RELEASE})
  842. set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries}
  843. debug ${BOOST_FILESYSTEM_LIB_RELEASE} optimized ${BOOST_FILESYSTEM_LIB_RELEASE})
  844. endif ()
  845. endif (WIN32)
  846. # 3-D
  847. if (WITH_3D_ADAM_MODEL)
  848. set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries}
  849. ${OPENGL_LIBRARIES} ${GLUT_LIBRARY} ${GLEW_LIBRARY} ${FREE_IMAGE_LIBRARY})
  850. endif (WITH_3D_ADAM_MODEL)
  851. if (WITH_3D_RENDERER)
  852. set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries} ${GLUT_LIBRARY} ${OPENGL_LIBRARIES})
  853. endif (WITH_3D_RENDERER)
  854. if (WITH_CERES)
  855. set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries} ${CERES_LIBRARIES})
  856. endif (WITH_CERES)
  857. if (WITH_FLIR_CAMERA)
  858. set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries} ${SPINNAKER_LIB})
  859. endif (WITH_FLIR_CAMERA)
  860. # Pthread
  861. if (UNIX OR APPLE)
  862. set(OpenPose_3rdparty_libraries ${OpenPose_3rdparty_libraries} pthread)
  863. endif (UNIX OR APPLE)
  864. set(examples_3rdparty_libraries ${OpenPose_3rdparty_libraries} ${GFLAGS_LIBRARY})
  865. ### ADD SUBDIRECTORIES
  866. if (Caffe_FOUND)
  867. add_subdirectory(src)
  868. if (BUILD_EXAMPLES)
  869. add_subdirectory(examples)
  870. endif (BUILD_EXAMPLES)
  871. endif (Caffe_FOUND)
  872. ### DOWNLOAD MODELS
  873. # Download the models if flag is set
  874. message(STATUS "Download the models.")
  875. # URL to the models
  876. set(OPENPOSE_URL "http://posefs1.perception.cs.cmu.edu/OpenPose/models/")
  877. download_model("BODY_25" ${DOWNLOAD_BODY_25_MODEL} pose/body_25/pose_iter_584000.caffemodel
  878. 78287B57CF85FA89C03F1393D368E5B7) # Body (BODY_25)
  879. download_model("body (COCO)" ${DOWNLOAD_BODY_COCO_MODEL} pose/coco/pose_iter_440000.caffemodel
  880. 5156d31f670511fce9b4e28b403f2939) # Body (COCO)
  881. download_model("body (MPI)" ${DOWNLOAD_BODY_MPI_MODEL} pose/mpi/pose_iter_160000.caffemodel
  882. 2ca0990c7562bd7ae03f3f54afa96e00) # Body (MPI)
  883. download_model("face" ${DOWNLOAD_FACE_MODEL} face/pose_iter_116000.caffemodel
  884. e747180d728fa4e4418c465828384333) # Face
  885. download_model("hand" ${DOWNLOAD_HAND_MODEL} hand/pose_iter_102000.caffemodel
  886. a82cfc3fea7c62f159e11bd3674c1531) # Hand
  887. message(STATUS "Models Downloaded.")
  888. ### PYTHON
  889. if (BUILD_PYTHON)
  890. if (WIN32)
  891. execute_process(COMMAND cmd /c cd ${CMAKE_SOURCE_DIR} & git submodule update --init 3rdparty/pybind11/)
  892. add_subdirectory(3rdparty/pybind11)
  893. add_subdirectory(python)
  894. elseif (UNIX OR APPLE)
  895. if (Caffe_FOUND)
  896. execute_process(COMMAND git submodule update --init ${CMAKE_SOURCE_DIR}/3rdparty/pybind11/)
  897. # execute_process(COMMAND git submodule update --init --recursive --remote) # This would initialize them all
  898. add_subdirectory(3rdparty/pybind11)
  899. add_subdirectory(python)
  900. endif (Caffe_FOUND)
  901. else (WIN32)
  902. message(FATAL_ERROR "Unknown OS.")
  903. endif (WIN32)
  904. endif (BUILD_PYTHON)
  905. ### GENERATE DOCUMENTATION
  906. if (UNIX OR APPLE)
  907. if (BUILD_DOCS)
  908. find_package(Doxygen)
  909. if (DOXYGEN_FOUND)
  910. # Set input and output files
  911. set(DOXYGEN_FILE ${CMAKE_SOURCE_DIR}/doc/doc_autogeneration.doxygen)
  912. # Custom target to build the documentation
  913. add_custom_target(doc_doxygen ALL
  914. COMMAND ${DOXYGEN_EXECUTABLE} ${DOXYGEN_FILE}
  915. WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}/doc
  916. COMMENT "Generating API documentation with Doxygen"
  917. VERBATIM)
  918. else (DOXYGEN_FOUND)
  919. message(FATAL_ERROR "Doxygen needs to be installed to generate the doxygen documentation.")
  920. endif (DOXYGEN_FOUND)
  921. endif (BUILD_DOCS)
  922. endif (UNIX OR APPLE)
  923. ### INSTALL
  924. if (UNIX OR APPLE)
  925. if (Caffe_FOUND)
  926. # Install the headers
  927. install(DIRECTORY ${CMAKE_SOURCE_DIR}/include/openpose DESTINATION include)
  928. install(EXPORT OpenPose DESTINATION lib/OpenPose)
  929. if (BUILD_CAFFE)
  930. install(DIRECTORY ${CMAKE_BINARY_DIR}/caffe/include/caffe DESTINATION include)
  931. install(DIRECTORY ${CMAKE_BINARY_DIR}/caffe/lib/ DESTINATION lib)
  932. endif (BUILD_CAFFE)
  933. # Compute installation prefix relative to this file
  934. configure_file(
  935. ${CMAKE_SOURCE_DIR}/cmake/OpenPoseConfig.cmake.in
  936. ${CMAKE_BINARY_DIR}/cmake/OpenPoseConfig.cmake @ONLY)
  937. install(FILES ${CMAKE_BINARY_DIR}/cmake/OpenPoseConfig.cmake
  938. DESTINATION lib/OpenPose)
  939. # Uninstall target
  940. configure_file(
  941. "${CMAKE_SOURCE_DIR}/cmake/cmake_uninstall.cmake.in"
  942. "${CMAKE_BINARY_DIR}/cmake_uninstall.cmake"
  943. IMMEDIATE @ONLY)
  944. add_custom_target(uninstall
  945. COMMAND ${CMAKE_COMMAND} -P ${CMAKE_CURRENT_BINARY_DIR}/cmake_uninstall.cmake)
  946. endif (Caffe_FOUND)
  947. endif (UNIX OR APPLE)
Tip!

Press p or to see the previous file or, n or to see the next file

Comments

Loading...