Duankaiwen / CenterNet

Codes for our paper "CenterNet: Keypoint Triplets for Object Detection" .

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

import top_pool error!!

Kangzf1996 opened this issue · comments

File "train.py", line 18, in
from nnet.py_factory import NetworkFactory
File "/home/travail/zfkang/CenterNet/nnet/py_factory.py", line 8, in
from models.py_utils.data_parallel import DataParallel
File "/home/travail/zfkang/CenterNet/models/py_utils/init.py", line 6, in
from ._cpools import TopPool, BottomPool, LeftPool, RightPool
File "/home/travail/zfkang/CenterNet/models/py_utils/_cpools/init.py", line 8, in
import top_pool, bottom_pool, left_pool, right_pool
ImportError: /usagers3/zhkan/.local/lib/python3.6/site-packages/cpools-0.0.0-py3.6-linux-x86_64.egg/top_pool.cpython-36m-x86_64-linux-gnu.so: undefined symbol: _ZNSt19basic_ostringstreamIcSt11char_traitsIcESaIcEEC1Ev

Can you help me to deal this error? Thank you!

the reason is the pytorch and cuda version, it works well now.

but if donot change the pytorch 1.2 version,how i can solve the problem?

but if donot change the pytorch 1.2 version,how i can solve the problem?

Hi there. I faced this problem before. I suspect that you have incomplete compilation of the corner pooling layers here:

cd <CenterNet dir>/models/py_utils/_cpools/

You can continue to use your Pytorch >= 1.0. I managed to correct this problem by following:

princeton-vl/CornerNet#33
or
https://github.com/princeton-vl/CornerNet/pull/65/files

Once you correct this problem, the corner pooling layers can be compiled properly, and you can run your training.

the reason is the pytorch and cuda version, it works well now.

which version does it work with ??

You can continue to use your Pytorch >= 1.0. I managed to correct this problem by following:

princeton-vl/CornerNet#33 or https://github.com/princeton-vl/CornerNet/pull/65/files

I tried this but I am still getting the same error, any solution for this?