faster_rcnn_resnet50

Faster R-CNN From Scratch (PyTorch)

A from-scratch PyTorch implementation of Faster R-CNN (Ren et al., 2015), aiming to closely — not exactly — reproduce paper-level results on Pascal VOC under real compute constraints.

Dataset

• Training: VOC2007 trainval + VOC2012 trainval ("07+12" protocol), 5,011 + 11,540 = 16,551 images, read from each dataset's ImageSets/Main/trainval.txt (not Segmentation or Layout — an early bug in this project pointed at the wrong subfolder and silently shrank the dataset to ~1,446 images).
• Evaluation: VOC2007 test. VOC2012 test is not used, since its ground truth requires official evaluation-server submission.

Backbone

• ResNet-50 (deviation from the paper's ResNet-101, for compute reasons — expect a modest mAP gap as a known, accepted trade-off).

• Initialized from ImageNet-pretrained weights.

Image Preprocessing

• Resize so the shorter side = 600px, longer side capped at 1000px, aspect ratio preserved (matches the paper's protocol; an earlier fixed-224×224 resize was replaced after confirming this in the paper text).
• Images are padded with max_dimensions in the batch to create a batch image tensor

Anchors

• 9 anchors per grid location: 3 scales (128², 256², 512²) × 3 ratios (1:1, 1:2, 2:1).

Boundary handling (train vs. test)

• Training: cross-boundary anchors are excluded entirely from the loss (not labeled positive/negative — ignored).
• Testing: no exclusion; decoded proposals are clipped to the image boundary instead.

Anchor Labeling (IoU-based)

Convention used: positive = 1, negative = -1, ignore = 0.

• Positive: (i) the anchor(s) with the highest IoU for a given GT box, OR (ii) any anchor with IoU > 0.7 with any GT box.
• Negative: IoU < 0.3 with all GT boxes.
• Ignore: neither of the above — excluded from the loss.
• Condition (i) is applied last, so it can override a negative/ignore label.

RPN Loss

• 256 anchors sampled per image, ~1:1 positive:negative ratio (padded with negatives if fewer than 128 positives are available).
• Classification: cross-entropy over the sampled anchors (2-class: background/foreground).
• Regression: Smooth L1 on (t_x, t_y, t_w, t_h) deltas, positive anchors only, normalized by positive count.
• Combined: loss = cls_loss + λ * reg_loss, λ = 10.

Proposal Generation (decode → clip → filter → NMS → top-N)

1. Decode anchors + predicted deltas → boxes
2. Clip to image boundary (test-time; training excludes cross-boundary anchors upstream instead)
3. Filter boxes smaller than 16px
4. Select Pre-NMS-Top-N Boxes
5. NMS, IoU threshold 0.7
6. Keep Post-NMS-Top-N Boxes

RoI Pooling

• RoIPool projects each proposal (corner format, absolute px) onto the shared feature map using stride = image_dim // feature_map_dim, then max-pools each projected region to a fixed 7×7 output — the original Fast R-CNN "RoI Pooling" (quantized max-pool over per-bin floor/ceil boundaries), not the later RoIAlign (Mask R-CNN's bilinear-interpolated variant).
• Two implementations of the same operation: a manual "loop" mode (default) that computes each bin's boundaries explicitly (every bin covers ≥1 pixel even when a proposal is smaller than the output size), and an "adaptive" mode via nn.AdaptiveMaxPool2d.

Detection Head (Fast R-CNN)

• Reuses conv5_x of an ImageNet-pretrained ResNet-50 as the region classifier — the shared backbone is split at conv4_x/conv5_x: conv4_x's output is the shared/RPN feature map, conv5_x becomes the per-RoI head. BatchNorm affine params are frozen and .train() is overridden to keep those BN layers in eval() mode (freezing requires_grad alone doesn't stop .train() from reactivating BN running-stat updates).
• Global average pool over conv5_x's output, then two sibling nn.Linear heads:
  • Classification: num_classes + 1 logits (VOC's 20 object classes + 1 background class).
  • Regression: num_classes * 4 box deltas — class-specific, unlike the RPN's class-agnostic deltas.

Detection Loss

• Classification: cross-entropy over the sampled proposals (21-way: 20 VOC classes + background).
• Regression: Smooth L1, computed only on the positive proposals' predicted deltas for their own ground-truth class, gathered out of the class-specific [N, num_classes, 4] delta tensor — summed, then divided by the number of sampled proposals for that image (not just the positive count).
• Regression targets are (t_x, t_y, t_w, t_h) deltas (same form as the RPN's), normalized by delta_std = (0.1, 0.1, 0.2, 0.2) — the Fast R-CNN paper's convention for zero-mean/unit-variance targets. Unlike the RPN's unnormalized deltas — any code decoding detection-head deltas back into boxes must multiply by delta_std first, or the decoded boxes come out silently near-zero-offset.
• Combined: loss = cls_loss + λ * reg_loss, λ = 1 (Fast R-CNN's default balancing weight — unlike the RPN's λ = 10).

Detection Net (Inference Decode)

• Wraps a trained DetectionHead for test-time use: given RegionProposalNetwork proposals and their RoIPool-ed features, runs the batched detection head once, then per image:
  1. Every (proposal, foreground class) pair whose softmax probability exceeds score_thresh (default 0.3) is emitted as a candidate — not just the argmax class. One proposal can therefore produce several detections, and a proposal whose highest-scoring class is background still contributes its foreground classes.
  2. Each candidate's box deltas for its own emitted class are gathered out of the class-specific delta tensor, un-normalized by delta_std, and decoded back to boxes with the same center-format inverse transform as the RPN's decoder.
  3. Boxes are clipped to the image's pre-padding size, then boxes that clipping collapsed to under min_box_size in either dimension are dropped.
  4. Per-class NMS (torchvision.ops.batched_nms, IoU nms_iou_thresh, default 0.3), then a top-max_detections_per_image cap by score (default 100).

Why argmax was replaced

The original decode kept only the argmax class per proposal and dropped the proposal when that was background. Measured on VOC2007 test, that emitted 16,819 detections against 14,976 GT boxes — 1.12 per object, where a standard Fast R-CNN emits 10–100× more — and capped mean recall at 0.613 while the RPN was supplying 80% proposal recall. Because 11-point AP scores p_interp(t) = 0 for every t above the achieved recall, mAP was pinned at 0.5389 against a ceiling of 0.6046 that the recall alone imposed; precision was already running at 89% of that ceiling. The loss was objects that never became detections at all, not objects ranked badly.

Per-class NMS (rather than class-agnostic) matters for the same metric: a person box must not suppress an overlapping horse box.

Training Protocol (4-Step Alternating Training, per the paper)

Step 1 hyperparameters (from the paper): SGD, momentum 0.9, weight decay 0.0005, lr 0.001 for the first ~60k mini-batches then 0.0001 for ~20k more (paper's batch-size-1 framing). This project's realized schedule: batch size 2 (a deliberate deviation for GPU throughput), 10 total epochs over 07+12 (~82,760 iterations) — 8 epochs at lr 0.001, 2 at lr 0.0001.

Results (VOC2007 test)

RPN Proposal Recall

Detection Net (Fast R-CNN head)

Per-class breakdown, score_thresh=0.3, nms_iou_thresh=0.3:

mAP @ IoU 0.5: 0.6252 — mean recall[-1]: 0.7485 — total detections: 44,520 — total GT (non-difficult): 12,032 — difficult GT excluded: 2,944.

Known Deviations From the Paper (Summary)

None of these were isolated by a controlled ablation — the ~63% mAP reflects their combined effect, not any single deviation's contribution.