fix bug open-mmlab#292

vernitgarg · May 22, 2020 · 96a172c · 96a172c
1 parent 6594153
commit 96a172c
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 11 deletions.
diff --git a/doc/GETTING_STARTED.md b/doc/GETTING_STARTED.md
@@ -27,13 +27,20 @@ d. Install mmskeleton:
 python setup.py develop
 ```
 
-e. [Optional] Install mmdetection for person detection:
+e. [Optional] Install nms for person estimation:
+``` shell
+cd mmskeleton/ops/nms/
+python setup_linux.py develop
+cd ../../../
+```
+
+f. [Optional] Install mmdetection for person detection:
 ``` shell
 python setup.py develop --mmdet
 ```
 In the event of a failure installation, please install [mmdetection](https://github.com/open-mmlab/mmdetection/blob/master/docs/INSTALL.md) manually.
 
-f. To verify that mmskeleton and mmdetection installed correctly, use:
+g. To verify that mmskeleton and mmdetection installed correctly, use:
 ```shell
 python mmskl.py pose_demo [--gpus $GPUS]
 # or "python mmskl.py pose_demo_HD [--gpus $GPUS]" for a higher accuracy

diff --git a/setup.py b/setup.py
@@ -177,15 +177,15 @@ def make_cuda_ext(name, module, sources, include_dirs=[]):
             'https://github.com/open-mmlab/mmdetection/tarball/v1.0rc1/#egg=mmdet-v1.0rc1'
         ],
         install_requires=install_requires,
-        ext_modules=[
-            make_cython_ext(name='cpu_nms',
-                            module='mmskeleton.ops.nms',
-                            sources=['cpu_nms.pyx']),
-            make_cuda_ext(name='gpu_nms',
-                          module='mmskeleton.ops.nms',
-                          sources=['nms_kernel.cu', 'gpu_nms.pyx'],
-                          include_dirs=[np.get_include()]),
-        ],
+        # ext_modules=[
+        #     make_cython_ext(name='cpu_nms',
+        #                     module='mmskeleton.ops.nms',
+        #                     sources=['cpu_nms.pyx']),
+        #     make_cuda_ext(name='gpu_nms',
+        #                   module='mmskeleton.ops.nms',
+        #                   sources=['nms_kernel.cu', 'gpu_nms.pyx'],
+        #                   include_dirs=[np.get_include()]),
+        # ],
         cmdclass={
             'build_ext': BuildExtension,
         },