GitHub - SirLPS/roi_pooling: four ways to compare the speed of roi pooling in pytorch

SirLPS / roi_pooling Public

Notifications You must be signed in to change notification settings
Fork 9
Star 62

four ways to compare the speed of roi pooling in pytorch

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 4 Commits
RoI_pooling		RoI_pooling
README		README
roi_cupy.py		roi_cupy.py
roi_module.py		roi_module.py
speed.py		speed.py
speed1.py		speed1.py

Repository files navigation

use_cuda: True, has_backward: True
method0: 0.0344547653198242, batch_size: 1, size: 50, num_rois: 300
method1: 0.1322056961059570, batch_size: 1, size: 50, num_rois: 300
method2: 0.1307379817962646, batch_size: 1, size: 50, num_rois: 300
method3: 0.2016681671142578, batch_size: 1, size: 50, num_rois: 300

# Referenc: https://github.com/blackyang/compare_roi_pooling_speed
# see also: http://www.cnblogs.com/king-lps/p/9026798.html