|
|
@@ -554,7 +554,7 @@ def linenet_newresnet50fpn(
|
|
|
|
|
|
backbone =resnet50fpn()
|
|
|
featmap_names=['0', '1', '2', '3','pool']
|
|
|
- print(f'featmap_names:{featmap_names}')
|
|
|
+ # print(f'featmap_names:{featmap_names}')
|
|
|
roi_pooler = MultiScaleRoIAlign(
|
|
|
featmap_names=featmap_names,
|
|
|
output_size=7,
|
|
|
@@ -562,14 +562,14 @@ def linenet_newresnet50fpn(
|
|
|
)
|
|
|
num_features=len(featmap_names)
|
|
|
anchor_sizes = tuple((int(16 * 2 ** i),) for i in range(num_features)) # 自动生成不同大小
|
|
|
- print(f'anchor_sizes:{anchor_sizes}')
|
|
|
+ # print(f'anchor_sizes:{anchor_sizes}')
|
|
|
aspect_ratios = ((0.5, 1.0, 2.0),) * num_features
|
|
|
- print(f'aspect_ratios:{aspect_ratios}')
|
|
|
+ # print(f'aspect_ratios:{aspect_ratios}')
|
|
|
|
|
|
|
|
|
anchor_generator = AnchorGenerator(sizes=anchor_sizes, aspect_ratios=aspect_ratios)
|
|
|
# anchors = anchor_generator.generate_anchors()
|
|
|
- print("Number of anchor sizes:", len(anchor_generator.sizes)) # 应为 5
|
|
|
+ # print("Number of anchor sizes:", len(anchor_generator.sizes)) # 应为 5
|
|
|
model = LineNet(backbone, num_classes=num_classes,anchor_generator=anchor_generator,
|
|
|
|
|
|
box_roi_pool=roi_pooler,
|
lstrlq