├── .gitignore ├── LICENSE ├── README.md ├── _assets └── prroi_visualization.png ├── pytorch ├── prroi_pool │ ├── .gitignore │ ├── __init__.py │ ├── functional.py │ ├── prroi_pool.py │ └── src │ │ ├── prroi_pooling_gpu.c │ │ ├── prroi_pooling_gpu.h │ │ ├── prroi_pooling_gpu_impl.cu │ │ └── prroi_pooling_gpu_impl.cuh └── tests │ └── test_prroi_pooling2d.py ├── src ├── prroi_pooling_gpu_impl.cu └── prroi_pooling_gpu_impl.cuh └── tensorflow ├── prroi_pool ├── CMakeLists.txt ├── __init__.py ├── precise_roi_pooling_ops.py └── src │ ├── kernels │ ├── build_cuda.py │ ├── external │ │ ├── prroi_pooling_gpu_impl.cu │ │ └── prroi_pooling_gpu_impl.cuh │ ├── precise_roi_pooling.h │ ├── precise_roi_pooling_kernels.cc │ └── precise_roi_pooling_kernels.cu.cc │ └── ops │ └── precise_roi_pooling_ops.cc └── tests ├── precise_roi_pooling_ops_test.py └── test_binaries └── 2_2_0.5 ├── features.npy ├── gradients0.npy ├── gradients1.npy ├── real_outputs.npy └── rois.npy /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vacancy/PreciseRoIPooling/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vacancy/PreciseRoIPooling/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vacancy/PreciseRoIPooling/HEAD/README.md -------------------------------------------------------------------------------- /_assets/prroi_visualization.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vacancy/PreciseRoIPooling/HEAD/_assets/prroi_visualization.png -------------------------------------------------------------------------------- /pytorch/prroi_pool/.gitignore: -------------------------------------------------------------------------------- 1 | *.o 2 | /_prroi_pooling 3 | -------------------------------------------------------------------------------- /pytorch/prroi_pool/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vacancy/PreciseRoIPooling/HEAD/pytorch/prroi_pool/__init__.py -------------------------------------------------------------------------------- /pytorch/prroi_pool/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vacancy/PreciseRoIPooling/HEAD/pytorch/prroi_pool/functional.py -------------------------------------------------------------------------------- /pytorch/prroi_pool/prroi_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vacancy/PreciseRoIPooling/HEAD/pytorch/prroi_pool/prroi_pool.py -------------------------------------------------------------------------------- /pytorch/prroi_pool/src/prroi_pooling_gpu.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vacancy/PreciseRoIPooling/HEAD/pytorch/prroi_pool/src/prroi_pooling_gpu.c -------------------------------------------------------------------------------- /pytorch/prroi_pool/src/prroi_pooling_gpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vacancy/PreciseRoIPooling/HEAD/pytorch/prroi_pool/src/prroi_pooling_gpu.h -------------------------------------------------------------------------------- /pytorch/prroi_pool/src/prroi_pooling_gpu_impl.cu: -------------------------------------------------------------------------------- 1 | ../../../src/prroi_pooling_gpu_impl.cu -------------------------------------------------------------------------------- /pytorch/prroi_pool/src/prroi_pooling_gpu_impl.cuh: -------------------------------------------------------------------------------- 1 | ../../../src/prroi_pooling_gpu_impl.cuh -------------------------------------------------------------------------------- /pytorch/tests/test_prroi_pooling2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vacancy/PreciseRoIPooling/HEAD/pytorch/tests/test_prroi_pooling2d.py -------------------------------------------------------------------------------- /src/prroi_pooling_gpu_impl.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vacancy/PreciseRoIPooling/HEAD/src/prroi_pooling_gpu_impl.cu -------------------------------------------------------------------------------- /src/prroi_pooling_gpu_impl.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vacancy/PreciseRoIPooling/HEAD/src/prroi_pooling_gpu_impl.cuh -------------------------------------------------------------------------------- /tensorflow/prroi_pool/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vacancy/PreciseRoIPooling/HEAD/tensorflow/prroi_pool/CMakeLists.txt -------------------------------------------------------------------------------- /tensorflow/prroi_pool/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vacancy/PreciseRoIPooling/HEAD/tensorflow/prroi_pool/__init__.py -------------------------------------------------------------------------------- /tensorflow/prroi_pool/precise_roi_pooling_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vacancy/PreciseRoIPooling/HEAD/tensorflow/prroi_pool/precise_roi_pooling_ops.py -------------------------------------------------------------------------------- /tensorflow/prroi_pool/src/kernels/build_cuda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vacancy/PreciseRoIPooling/HEAD/tensorflow/prroi_pool/src/kernels/build_cuda.py -------------------------------------------------------------------------------- /tensorflow/prroi_pool/src/kernels/external/prroi_pooling_gpu_impl.cu: -------------------------------------------------------------------------------- 1 | ../../../../../src/prroi_pooling_gpu_impl.cu -------------------------------------------------------------------------------- /tensorflow/prroi_pool/src/kernels/external/prroi_pooling_gpu_impl.cuh: -------------------------------------------------------------------------------- 1 | ../../../../../src/prroi_pooling_gpu_impl.cuh -------------------------------------------------------------------------------- /tensorflow/prroi_pool/src/kernels/precise_roi_pooling.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vacancy/PreciseRoIPooling/HEAD/tensorflow/prroi_pool/src/kernels/precise_roi_pooling.h -------------------------------------------------------------------------------- /tensorflow/prroi_pool/src/kernels/precise_roi_pooling_kernels.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vacancy/PreciseRoIPooling/HEAD/tensorflow/prroi_pool/src/kernels/precise_roi_pooling_kernels.cc -------------------------------------------------------------------------------- /tensorflow/prroi_pool/src/kernels/precise_roi_pooling_kernels.cu.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vacancy/PreciseRoIPooling/HEAD/tensorflow/prroi_pool/src/kernels/precise_roi_pooling_kernels.cu.cc -------------------------------------------------------------------------------- /tensorflow/prroi_pool/src/ops/precise_roi_pooling_ops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vacancy/PreciseRoIPooling/HEAD/tensorflow/prroi_pool/src/ops/precise_roi_pooling_ops.cc -------------------------------------------------------------------------------- /tensorflow/tests/precise_roi_pooling_ops_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vacancy/PreciseRoIPooling/HEAD/tensorflow/tests/precise_roi_pooling_ops_test.py -------------------------------------------------------------------------------- /tensorflow/tests/test_binaries/2_2_0.5/features.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vacancy/PreciseRoIPooling/HEAD/tensorflow/tests/test_binaries/2_2_0.5/features.npy -------------------------------------------------------------------------------- /tensorflow/tests/test_binaries/2_2_0.5/gradients0.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vacancy/PreciseRoIPooling/HEAD/tensorflow/tests/test_binaries/2_2_0.5/gradients0.npy -------------------------------------------------------------------------------- /tensorflow/tests/test_binaries/2_2_0.5/gradients1.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vacancy/PreciseRoIPooling/HEAD/tensorflow/tests/test_binaries/2_2_0.5/gradients1.npy -------------------------------------------------------------------------------- /tensorflow/tests/test_binaries/2_2_0.5/real_outputs.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vacancy/PreciseRoIPooling/HEAD/tensorflow/tests/test_binaries/2_2_0.5/real_outputs.npy -------------------------------------------------------------------------------- /tensorflow/tests/test_binaries/2_2_0.5/rois.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vacancy/PreciseRoIPooling/HEAD/tensorflow/tests/test_binaries/2_2_0.5/rois.npy --------------------------------------------------------------------------------