FasterRCNN and ROIAlign are Non-Functional in Master

[VarunS1997]

Current Behavior:

While investigating #2388, I learned that the current tests for FasterRCNN all fail. Further, the functionality of ROIAlign doesn't appear to be working as is. Further, ROIAlign may be ready for deletion as per TODO.

Expected Behavior:

Tests for FasterRCNN as well as a basic usage of ROIAlign should both work.

Steps To Reproduce:

Run FasterRCNN tests

Version:

HEAD

Anything else:

N/A

[sineeli]

@VarunS1997

Seems to be ops.take missing axis argument in ROIAlign code so tried it and it seems to be working as expected.

Compared same with Tensorflow Model Garden as the pipelines of MaskRCNN and FasterRCNN are stable in TFM.

Tested gist