PSROIAlign:(面向)Tensorflow C ++ API中的PsRoIAlign操作

时间:2024-06-05 08:49:46
【文件属性】:

文件名称:PSROIAlign:(面向)Tensorflow C ++ API中的PsRoIAlign操作

文件大小:47KB

文件格式:ZIP

更新时间:2024-06-05 08:49:46

cpp tensorflow faster-rcnn objectdetection roipooling

Tensorflow C ++ API中的PsRoIAlign操作 PsRoIAlign涉及用于 (位置敏感的池操作)的插值技术,该插值思想是在提出的,以避免对进行任何量化。 PsRoIAlign的首次采用可能是在本文中的。 您可以找到有关的RoiPooling技术和更多详细信息。 该存储库包含Tensorflow C ++ API中PsRoIAlign操作的实现代码。 您可以在许多流行的两阶段对象检测器中使用此操作。 既欢迎使用PsRoIAlign进行的研究工作,也欢迎对该存储库做出贡献。 要在自己的计算机上使用此op,只需执行以下步骤: 将头文件“ cuda_config.h”从“ your_python_path / site-packages / external / local_config_cuda / cuda / cuda / cuda_config.h”复制到“


【文件预览】:
PSROIAlign-master
----.gitignore(270B)
----ps_roi_align_op.cu(9KB)
----common.h(2KB)
----build()
--------test_op.sh(90B)
----ps_roi_align_grad_op.cc(25KB)
----1.4()
--------cuda_config.h(964B)
----LICENSE(1KB)
----common.cc(2KB)
----rotated_ps_roi_align_grad_op.cu(12KB)
----README.md(3KB)
----ps_roi_align_op.h(3KB)
----rotated_ps_roi_align_grad_op.cc(30KB)
----ps_roi_align_grad_op.cu(10KB)
----test_op.py(12KB)
----rotated_ps_roi_align_op.cu(12KB)
----ps_roi_align_op.cc(15KB)
----rotated_ps_roi_align_op.h(3KB)
----CMakeLists.txt(4KB)
----work_sharder.h(2KB)
----rotated_ps_roi_align_op.cc(22KB)

网友评论