object detection: ROI Align for Faster R-CNN

tensorflow / models

Models and examples built with TensorFlow

Other

77.18k stars 45.75k forks source link

object detection: ROI Align for Faster R-CNN #3727

Closed andyguo007 closed 6 years ago

andyguo007 commented 6 years ago

Hi,

Do you have the plan to implement a ROIAlign layer (instead of ROI Pooliing) for Faster R-CNN?

Thanks,

tensorflowbutler commented 6 years ago

Thank you for your post. We noticed you have not filled out the following field in the issue template. Could you update them if they are relevant in your case, or leave them as N/A? Thanks. What is the top-level directory of the model you are using Have I written custom code OS Platform and Distribution TensorFlow installed from TensorFlow version Bazel version CUDA/cuDNN version GPU model and memory Exact command to reproduce

jianlong-yuan commented 6 years ago

Any news

ppwwyyxx commented 6 years ago

TF already has enough builtin ops to implement ROIAlign: https://github.com/tensorpack/tensorpack/blob/3e0dffaccdb6a36490970014570b5e7426b55bf5/examples/FasterRCNN/model_box.py#L156-L172

wt-huang commented 6 years ago

roi_align can be leveraged for Faster R-CNN.