From 7be71b02e2013a2b4be32511002f1ea6987afaca Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 11 Apr 2020 10:56:20 -0700 Subject: [PATCH] get_yolo_layers() --- models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models.py b/models.py index 90d94ebe..dea75221 100755 --- a/models.py +++ b/models.py @@ -334,7 +334,7 @@ class Darknet(nn.Module): def get_yolo_layers(model): - return [i for i, m in enumerate(model.module_list) if m.__class__.__name__ == 'YOLOLayer'] # [89, 101, 1113] + return [i for i, m in enumerate(model.module_list) if m.__class__.__name__ == 'YOLOLayer'] # [89, 101, 113] def load_darknet_weights(self, weights, cutoff=-1):