def forward(self, x):
    if (x.dim() == 4):
        assert (list(x.shape[2:]) == [1, 1])
    x = x.flatten(start_dim=1)
    scores = self.cls_score(x)
    bbox_deltas = self.bbox_pred(x)
    return (scores, bbox_deltas)
def forward(self, x):
    if (x.dim() == 4):
        assert (list(x.shape[2:]) == [1, 1])
    x = x.flatten(start_dim=1)
    scores = self.cls_score(x)
    bbox_deltas = self.bbox_pred(x)
    return (scores, bbox_deltas)