├── .gitignore ├── README.md ├── datasets └── linemod │ └── dataset_posecnn.py ├── eval_net.py ├── lib ├── HoughVoting │ ├── .gitignore │ ├── __init__.py │ ├── houghvoting.cc │ ├── houghvoting.py │ └── setup.py ├── center_est_funcs.py ├── loss_funcions.py ├── roi_pool_pytorch.py ├── vgg16_convs.py ├── vgg16_convs_combine_mask.py └── vgg16_convs_combine_seg_center.py ├── testpytorch.py └── train_net.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nxu96/PoseCNN_PyTorch/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nxu96/PoseCNN_PyTorch/HEAD/README.md -------------------------------------------------------------------------------- /datasets/linemod/dataset_posecnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nxu96/PoseCNN_PyTorch/HEAD/datasets/linemod/dataset_posecnn.py -------------------------------------------------------------------------------- /eval_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nxu96/PoseCNN_PyTorch/HEAD/eval_net.py -------------------------------------------------------------------------------- /lib/HoughVoting/.gitignore: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/HoughVoting/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nxu96/PoseCNN_PyTorch/HEAD/lib/HoughVoting/__init__.py -------------------------------------------------------------------------------- /lib/HoughVoting/houghvoting.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nxu96/PoseCNN_PyTorch/HEAD/lib/HoughVoting/houghvoting.cc -------------------------------------------------------------------------------- /lib/HoughVoting/houghvoting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nxu96/PoseCNN_PyTorch/HEAD/lib/HoughVoting/houghvoting.py -------------------------------------------------------------------------------- /lib/HoughVoting/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nxu96/PoseCNN_PyTorch/HEAD/lib/HoughVoting/setup.py -------------------------------------------------------------------------------- /lib/center_est_funcs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nxu96/PoseCNN_PyTorch/HEAD/lib/center_est_funcs.py -------------------------------------------------------------------------------- /lib/loss_funcions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nxu96/PoseCNN_PyTorch/HEAD/lib/loss_funcions.py -------------------------------------------------------------------------------- /lib/roi_pool_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nxu96/PoseCNN_PyTorch/HEAD/lib/roi_pool_pytorch.py -------------------------------------------------------------------------------- /lib/vgg16_convs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nxu96/PoseCNN_PyTorch/HEAD/lib/vgg16_convs.py -------------------------------------------------------------------------------- /lib/vgg16_convs_combine_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nxu96/PoseCNN_PyTorch/HEAD/lib/vgg16_convs_combine_mask.py -------------------------------------------------------------------------------- /lib/vgg16_convs_combine_seg_center.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nxu96/PoseCNN_PyTorch/HEAD/lib/vgg16_convs_combine_seg_center.py -------------------------------------------------------------------------------- /testpytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nxu96/PoseCNN_PyTorch/HEAD/testpytorch.py -------------------------------------------------------------------------------- /train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nxu96/PoseCNN_PyTorch/HEAD/train_net.py --------------------------------------------------------------------------------