资源分类
》 资源 》算法》Rotated_ROIAlign

Rotated_ROIAlign

2020-02-26 | |  101 |   0 |   0

Rotated_ROIAlign

Rotated RoIAlign in pytorch, implemented with CUDA. This implementation is edited from caffe operator and use pytorch's data interface.

Usage

  1. First complile the source file

    python setup.py build_ext --inplace

  1. Then you can call the Rotated RoIAlign in python.

    pooler_rotated=ROIAlignRotated((32,192), spatial_scale = (1.), sampling_ratio = 0)
    image_roi_bbox=pooler_rotated(<Input image>,<Input RoI>)

Example

test.py is a example how to use the rotated RoIAlign. The input image


上一篇：RoIAlign-RoIPool-pytorch

下一篇：roialign_explode_issue

用户评价
全部评价

热门资源

  • allennlp-server

    allennlp-server Serve allennlp services as sep...

  • ubuntu-allennlp

    ubuntu-allennlp AllenAI AllenNLP image based o...

  • allennlp_extras

    allennlp_extras Some utilities build on top of...

  • allennlp-dureader

    An Apache 2.0 NLP research library, built on Py...

  • seetafaceJNI

    项目介绍 基于中科院seetaface2进行封装的JAVA...