Comments (3)
hi, @LouisNUST
Have you build your own NMS extension? See: prerequisites-NMS extension. And please ensure using the same cuda version with pytorch.
from multiposenet.pytorch.
@LiMeng95 Thanks for your reply.
I have precompiled my own NMS extentions:
Compiling nms kernels by nvcc...
Including CUDA code.
/home/louis/Documents/DetPose/MultiPoseNet/lib/nms
generating /tmp/tmpp59_1z1x/nms.c
setting the current directory to '/tmp/tmpp59_1z1x'
running build_ext
building 'nms' extension
creating home
creating home/louis
creating home/louis/Documents
creating home/louis/Documents/DetPose
creating home/louis/Documents/DetPose/MultiPoseNet
creating home/louis/Documents/DetPose/MultiPoseNet/lib
creating home/louis/Documents/DetPose/MultiPoseNet/lib/nms
creating home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src
gcc -pthread -B /home/louis/anaconda3/envs/MultiPose/compiler_compat -Wl,--sysroot=/ -Wsign-compare -DNDEBUG -g -fwrapv -O3 -Wall -Wstrict-prototypes -fPIC -DWITH_CUDA -I/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include -I/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH -I/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/THC -I/usr/local/cuda/include -I/home/louis/anaconda3/envs/MultiPose/include/python3.6m -c nms.c -o ./nms.o -std=c99
gcc -pthread -B /home/louis/anaconda3/envs/MultiPose/compiler_compat -Wl,--sysroot=/ -Wsign-compare -DNDEBUG -g -fwrapv -O3 -Wall -Wstrict-prototypes -fPIC -DWITH_CUDA -I/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include -I/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH -I/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/THC -I/usr/local/cuda/include -I/home/louis/anaconda3/envs/MultiPose/include/python3.6m -c /home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms.c -o ./home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms.o -std=c99
In file included from /home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH/TH.h:4:0,
from /home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms.c:1:
/home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms.c: In function ‘cpu_nms’:
/home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms.c:7:42: warning: passing argument 1 of ‘THLongTensor_isContiguous’ from incompatible pointer type [-Wincompatible-pointer-types]
THArgCheck(THLongTensor_isContiguous(boxes), 2, "boxes must be contiguous");
^
/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH/THGeneral.h:100:35: note: in definition of macro ‘THArgCheck’
THArgCheck(FILE, LINE, VA_ARGS);
^~~~~~~~~~~
/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH/THTensor.h:8:39: note: expected ‘const THLongTensor * {aka const struct THLongTensor *}’ but argument is of type ‘THFloatTensor * {aka struct THFloatTensor *}’
#define THTensor(NAME) TH_CONCAT_4(TH,Real,Tensor,NAME)
^
/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH/THGeneral.h:144:37: note: in definition of macro ‘TH_CONCAT_4_EXPAND’
#define TH_CONCAT_4_EXPAND(x,y,z,w) x ## y ## z ## w
^
/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH/THTensor.h:8:27: note: in expansion of macro ‘TH_CONCAT_4’
#define THTensor(NAME) TH_CONCAT_4(TH,Real,Tensor,NAME)
^~~~~~~~~~~
/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH/generic/THTensor.h:114:12: note: in expansion of macro ‘THTensor’
TH_API int THTensor_(isContiguous)(const THTensor self);
^~~~~~~~~
/home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms.c:9:42: warning: passing argument 1 of ‘THLongTensor_isContiguous’ from incompatible pointer type [-Wincompatible-pointer-types]
THArgCheck(THLongTensor_isContiguous(areas), 4, "areas must be contiguous");
^
/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH/THGeneral.h:100:35: note: in definition of macro ‘THArgCheck’
THArgCheck(FILE, LINE, VA_ARGS);
^~~~~~~~~~~
/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH/THTensor.h:8:39: note: expected ‘const THLongTensor * {aka const struct THLongTensor *}’ but argument is of type ‘THFloatTensor * {aka struct THFloatTensor *}’
#define THTensor(NAME) TH_CONCAT_4(TH,Real,Tensor_,NAME)
^
/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH/THGeneral.h:144:37: note: in definition of macro ‘TH_CONCAT_4_EXPAND’
#define TH_CONCAT_4_EXPAND(x,y,z,w) x ## y ## z ## w
^
/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH/THTensor.h:8:27: note: in expansion of macro ‘TH_CONCAT_4’
#define THTensor_(NAME) TH_CONCAT_4(TH,Real,Tensor_,NAME)
^~~~~~~~~~~
/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH/generic/THTensor.h:114:12: note: in expansion of macro ‘THTensor_’
TH_API int THTensor_(isContiguous)(const THTensor self);
^~~~~~~~~
gcc -pthread -B /home/louis/anaconda3/envs/MultiPose/compiler_compat -Wl,--sysroot=/ -Wsign-compare -DNDEBUG -g -fwrapv -O3 -Wall -Wstrict-prototypes -fPIC -DWITH_CUDA -I/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include -I/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/TH -I/home/louis/anaconda3/envs/pytorch36/lib/python3.6/site-packages/torch/utils/ffi/../../lib/include/THC -I/usr/local/cuda/include -I/home/louis/anaconda3/envs/MultiPose/include/python3.6m -c /home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms_cuda.c -o ./home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms_cuda.o -std=c99
/home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms_cuda.c: In function ‘gpu_nms’:
/home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms_cuda.c:29:35: warning: initialization from incompatible pointer type [-Wincompatible-pointer-types]
unsigned long long mask_flat = THCudaLongTensor_data(state, mask);
^~~~~~~~~~~~~~~~~~~~~
/home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms_cuda.c:37:40: warning: initialization from incompatible pointer type [-Wincompatible-pointer-types]
unsigned long long * mask_cpu_flat = THLongTensor_data(mask_cpu);
^~~~~~~~~~~~~~~~~
/home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms_cuda.c:40:39: warning: initialization from incompatible pointer type [-Wincompatible-pointer-types]
unsigned long long remv_cpu_flat = THLongTensor_data(remv_cpu);
^~~~~~~~~~~~~~~~~
/home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms_cuda.c:23:7: warning: unused variable ‘boxes_dim’ [-Wunused-variable]
int boxes_dim = THCudaTensor_size(state, boxes, 1);
^~~~~~~~~
gcc -pthread -shared -B /home/louis/anaconda3/envs/MultiPose/compiler_compat -L/home/louis/anaconda3/envs/MultiPose/lib -Wl,-rpath=/home/louis/anaconda3/envs/MultiPose/lib -Wl,--no-as-needed -Wl,--sysroot=/ ./_nms.o ./home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms.o ./home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/nms_cuda.o /home/louis/Documents/DetPose/MultiPoseNet/lib/nms/src/cuda/nms_kernel.cu.o -o ./_nms.so
the CUDA version is 10.1
Ubuntu: 18.04
python: 3.6.8
pytorch: 0.4.1
cuda90: 1.0
from multiposenet.pytorch.
Maybe the mismatch of CUDA version caused this problem. I used CUDA9.0, so that you can find package cuda90: 1.0
in multipose_environment.yaml
from multiposenet.pytorch.
Related Issues (20)
- I want to know the best result of this work? HOT 4
- Questions about key point network training HOT 12
- Missing license HOT 1
- Questions about training and evaluate HOT 1
- Improvements detail HOT 2
- this the numpy==1.14.3 and pycocotools==2.0.0 is ok? i have this error! HOT 3
- multipose_environment.yaml file HOT 4
- some questions about keypoints training? HOT 3
- PRN when using tester HOT 2
- Question about the baseline model HOT 20
- Build the NMS extension HOT 7
- No such file or directory: '/data/COCO/COCO.json' HOT 1
- 有关intermediate supervision的问题 HOT 2
- different input size of the two subnet when training HOT 10
- why weight_decay = 0? HOT 2
- detection subnet training HOT 9
- Maybe some bugs in network?
- Original image and heatmap alignment
- Details of Baseline model
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from multiposenet.pytorch.