Remove sample_conv as layer do not exist when users are building for …

…custom models
WenheLI · Aug 6, 2020 · 0a4cab3 · 0a4cab3
1 parent 2eeb08d
commit 0a4cab3
Showing 1 changed file with 0 additions and 15 deletions.
diff --git a/tf2_yolov4/tools/weights.py b/tf2_yolov4/tools/weights.py
@@ -29,10 +29,6 @@ def load_darknet_weights_in_yolo(yolo_model, darknet_weights_path):
     Returns:
         tf.keras.Model: YOLOv4 model with Darknet weights loaded.
     """
-    sample_conv_weights = (
-        yolo_model.get_layer("CSPDarknet53").get_layer("conv2d_32").get_weights()[0]
-    )
-
     model_layers = (
         yolo_model.get_layer("CSPDarknet53").layers
         + yolo_model.get_layer("YOLOv4_neck").layers
@@ -108,17 +104,6 @@ def load_darknet_weights_in_yolo(yolo_model, darknet_weights_path):
     darknet_weight_file.close()
     assert remaining_chars == 0
 
-    # Check if weights have been updated
-    sample_conv_weights_after_loading = (
-        yolo_model.get_layer("CSPDarknet53").get_layer("conv2d_32").get_weights()[0]
-    )
-    np.testing.assert_raises(
-        AssertionError,
-        np.testing.assert_array_equal,
-        sample_conv_weights,
-        sample_conv_weights_after_loading,
-    )
-
     return yolo_model