search

j96w / DenseFusion

"DenseFusion: 6D Object Pose Estimation by Iterative Dense Fusion" code repository
https://sites.google.com/view/densefusion
MIT License
1.08k stars 301 forks source link

有没有支持PyTorch1.5的版本？ #200

Open zhanghaobucunzai opened 2 years ago

kuaitoukid commented 2 years ago

You may refer this https://github.com/kuaitoukid/DenseFusion, I replace the KNN with a pytorch version, but sacrificed the speed.

wangqingyu985 commented 1 year ago

I used your repo and encountered this problem

Traceback (most recent call last): File "/home/wangqingyu/桌面/DenseFusion/tools/train.py", line 26, in from lib.loss import Loss File "/home/wangqingyu/桌面/DenseFusion/lib/loss.py", line 9, in from lib.knn.init import KNearestNeighbor File "/home/wangqingyu/桌面/DenseFusion/lib/knn/init.py", line 7, in from lib.knn import knn_pytorch as knn_pytorch File "/home/wangqingyu/桌面/DenseFusion/lib/knn/knn_pytorch/init.py", line 1, in from torch.utils.ffi import _wrap_function File "/home/wangqingyu/anaconda3/envs/new/lib/python3.8/site-packages/torch/utils/ffi/init.py", line 1, in raise ImportError("torch.utils.ffi is deprecated. Please use cpp extensions instead.") ImportError: torch.utils.ffi is deprecated. Please use cpp extensions instead.