Skip to content

S
seminar-breakout
Commit 480bc6e6 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix #987

parent cc3de54d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/FasterRCNN/train.py
View file @ 480bc6e6
...@@ -171,7 +171,7 @@ class ResNetC4Model(DetectionModel): ...@@ -171,7 +171,7 @@ class ResNetC4Model(DetectionModel):
feature_gap = GlobalAvgPooling('gap', feature_fastrcnn, data_format='channels_first') feature_gap = GlobalAvgPooling('gap', feature_fastrcnn, data_format='channels_first')
fastrcnn_label_logits, fastrcnn_box_logits = fastrcnn_outputs('fastrcnn', feature_gap, cfg.DATA.NUM_CLASS) fastrcnn_label_logits, fastrcnn_box_logits = fastrcnn_outputs('fastrcnn', feature_gap, cfg.DATA.NUM_CLASS)
fastrcnn_head = FastRCNNHead(proposals, fastrcnn_box_logits, fastrcnn_label_logits, fastrcnn_head = FastRCNNHead(proposals, fastrcnn_box_logits, fastrcnn_label_logits, gt_boxes,
tf.constant(cfg.FRCNN.BBOX_REG_WEIGHTS, dtype=tf.float32)) tf.constant(cfg.FRCNN.BBOX_REG_WEIGHTS, dtype=tf.float32))
if self.training: if self.training:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment