From 7357e40b8877ee4f0245a7c13a93361df9c92e33 Mon Sep 17 00:00:00 2001
From: Jon Crall <erotemic@gmail.com>
Date: Thu, 10 Oct 2019 04:56:27 -0400
Subject: [PATCH] Use standard NCHW instead of BCHW (#1516)
---
mmdet/models/detectors/cascade_rcnn.py | 4 ++--
mmdet/models/detectors/two_stage.py | 2 +-
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/mmdet/models/detectors/cascade_rcnn.py b/mmdet/models/detectors/cascade_rcnn.py
index 3e2d1af..86e971b 100644
--- a/mmdet/models/detectors/cascade_rcnn.py
+++ b/mmdet/models/detectors/cascade_rcnn.py
@@ -159,7 +159,7 @@ class CascadeRCNN(BaseDetector, RPNTestMixin):
proposals=None):
"""
Args:
- img (Tensor): of shape (B, C, H, W) encoding input images.
+ img (Tensor): of shape (N, C, H, W) encoding input images.
Typically these should be mean centered and std scaled.
img_meta (list[dict]): list of image info dict where each dict has:
@@ -301,7 +301,7 @@ class CascadeRCNN(BaseDetector, RPNTestMixin):
"""Run inference on a single image.
Args:
- img (Tensor): must be in shape (B, C, H, W)
+ img (Tensor): must be in shape (N, C, H, W)
img_meta (list[dict]): a list with one dictionary element.
See `mmdet/datasets/pipelines/formatting.py:Collect` for
details of meta dicts.
diff --git a/mmdet/models/detectors/two_stage.py b/mmdet/models/detectors/two_stage.py
index f35d25b..e457ea3 100644
--- a/mmdet/models/detectors/two_stage.py
+++ b/mmdet/models/detectors/two_stage.py
@@ -137,7 +137,7 @@ class TwoStageDetector(BaseDetector, RPNTestMixin, BBoxTestMixin,
proposals=None):
"""
Args:
- img (Tensor): of shape (B, C, H, W) encoding input images.
+ img (Tensor): of shape (N, C, H, W) encoding input images.
Typically these should be mean centered and std scaled.
img_meta (list[dict]): list of image info dict where each dict has:
--
GitLab