updates

2019-08-03 14:22:25 +02:00
parent 90daf8f19c
commit cd1f1eeecc
2 changed files with 2 additions and 2 deletions
@@ -338,7 +338,7 @@ def build_targets(model, targets):
    nt = len(targets)
    txy, twh, tcls, tbox, indices, anchor_vec = [], [], [], [], [], []
    for i in model.yolo_layers:
-        layer = model.module_list[i][0]
+        layer = model.module_list[i]

        # iou of targets-anchors
        t, a = targets, []