瀏覽代碼

Merge pull request #857 from will-jl944/develop_jf

Fix bug that occurs when `fpn_num_channels` is changed in FasterRCNN
FlyingQianMM 4 年之前
父節點
當前提交
9f882f6171
共有 2 個文件被更改,包括 2 次插入2 次删除
  1. 0 1
      dygraph/paddlex/cv/datasets/voc.py
  2. 2 1
      dygraph/paddlex/cv/models/detector.py

+ 0 - 1
dygraph/paddlex/cv/datasets/voc.py

@@ -14,7 +14,6 @@
 
 
 from __future__ import absolute_import
 from __future__ import absolute_import
 import copy
 import copy
-import os
 import os.path as osp
 import os.path as osp
 import random
 import random
 import re
 import re

+ 2 - 1
dygraph/paddlex/cv/models/detector.py

@@ -848,7 +848,8 @@ class FasterRCNN(BaseDetector):
                 if test_pre_nms_top_n is None else test_pre_nms_top_n,
                 if test_pre_nms_top_n is None else test_pre_nms_top_n,
                 'post_nms_top_n': test_post_nms_top_n
                 'post_nms_top_n': test_post_nms_top_n
             }
             }
-            head = ppdet.modeling.TwoFCHead(out_channel=1024)
+            head = ppdet.modeling.TwoFCHead(
+                in_channel=neck.out_shape[0].channels, out_channel=1024)
             roi_extractor_cfg = {
             roi_extractor_cfg = {
                 'resolution': 7,
                 'resolution': 7,
                 'spatial_scale': [1. / i.stride for i in neck.out_shape],
                 'spatial_scale': [1. / i.stride for i in neck.out_shape],