Open universewill opened 6 years ago
the code is not guaranteed to work with Python 3
In case someone might find it useful, I just created a fork with a branch called python3, containing the changes I needed to make to start training using python3:
https://github.com/landersson/DensePose/tree/python3
This is not well tested, but at least train_net.py seems to be working...
In case someone might find it useful, I just created a fork with a branch called python3, containing the changes I needed to make to start training using python3:
https://github.com/landersson/DensePose/tree/python3
This is not well tested, but at least train_net.py seems to be working...
thanks for the code,I compile it successfully,but when run the code ,there are som problm:
CUDA_VISIBLE_DEVICES=3 python tools/infer_simple.py \ --cfg configs/DensePose_ResNet101_FPN_s1x-e2e.yaml \ --output-dir DensePoseData/infer_out/ \ --image-ext jpg \ --wts https://s3.amazonaws.com/densepose/DensePose_ResNet101_FPN_s1x-e2e.pkl \ DensePoseData/demo_data/demo_im.jpg
Found Detectron ops lib: /root/anaconda3/envs/AIdance/lib/libcaffe2_detectron_ops_gpu.so
E1229 18:38:25.510565 119887 init_intrinsics_check.cc:43] CPU feature avx is present on your machine, but the Caffe2 binary is not compiled with it. It means you may not get the full speed of your CPU.
E1229 18:38:25.510648 119887 init_intrinsics_check.cc:43] CPU feature avx2 is present on your machine, but the Caffe2 binary is not compiled with it. It means you may not get the full speed of your CPU.
E1229 18:38:25.510828 119887 init_intrinsics_check.cc:43] CPU feature fma is present on your machine, but the Caffe2 binary is not compiled with it. It means you may not get the full speed of your CPU.
WARNING cnn.py: 25: [====DEPRECATE WARNING====]: you are creating an object from CNNModelHelper class which will be deprecated soon. Please use ModelHelper object with brew module. For more information, please refer to caffe2.ai and python/brew.py, python/brew_test.py for more information.
INFO net.py: 51: Loading weights from: /root/dance/densepose/model/DensePose_ResNet101_FPN_s1x-e2e.pkl
I1229 18:38:27.143235 119887 net_dag_utils.cc:102] Operator graph pruning prior to chain compute took: 0.000185281 secs
I1229 18:38:27.143592 119887 net_dag.cc:46] Number of parallel execution chains 36 Number of operators = 371
I1229 18:38:27.162335 119887 net_dag_utils.cc:102] Operator graph pruning prior to chain compute took: 0.000172346 secs
I1229 18:38:27.162598 119887 net_dag.cc:46] Number of parallel execution chains 30 Number of operators = 358
I1229 18:38:27.164830 119887 net_dag_utils.cc:102] Operator graph pruning prior to chain compute took: 1.9383e-05 secs
I1229 18:38:27.164917 119887 net_dag.cc:46] Number of parallel execution chains 10 Number of operators = 30
INFO infer_simple.py: 106: Processing DensePoseData/demo_data/demo_im.jpg -> DensePoseData/infer_out/demo_im.jpg.pdf
E1229 18:38:28.133205 119989 net_dag.cc:195] Exception from operator chain starting at '' (type 'Conv'): caffe2::EnforceNotMet: [enforce fail at blob.h:84] IsType359
in network generalized_rcnn
in exception above (most recent call last):
WARNING workspace.py: 190: File "tools/infer_simple.py", line 143, in
Do you have any advice ? thanks!
Is this support python3? I notice that the demo cmd is all python2.