Update

configuration.py

@@ -22,7 +22,7 @@ class Config:
 
   model_id = 'a0'
   detector_path = 'weights/yolov8n.pt'
-  classifier_path = 'weights/classifier-8-epoch10.keras'
+  classifier_path = 'weights/classifier-16-epoch9.keras'
   num_classes = len(id_to_name)
   input_shape = (1, num_frames, resolution, resolution, 3)
 

model.py

@@ -31,10 +31,10 @@ def build_movinet(output_size, config: Config):
   return model
 
 def build_classifier_head(input_size, config: Config):
-  # inputs = keras.Input(shape=(input_size,))
-  # classifier = AttentionDenseClassifierHead(2, config.num_classes)(inputs)
-  # model = keras.Model(inputs=inputs, outputs=classifier)
-  return keras.layers.Dense(config.num_classes, activation='softmax')
+  inputs = keras.Input(shape=(input_size,))
+  classifier = AttentionDenseClassifierHead(2, config.num_classes)(inputs)
+  model = keras.Model(inputs=inputs, outputs=classifier)
+  return model
 
 def build_model(movinet, classifier_head):
   return keras.models.Sequential([movinet, classifier_head])

weights/{classifier-8-epoch10.keras → classifier-16-epoch9.keras}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c11a51f095a4902755f375740980537257764536bd089d2ae7cbe1cbb8343184
-size 38477915
+oid sha256:822441184b3bc06eba5a6dfe82e8c7a5a1a07e5176723ea41c55f9b2bd63d547
+size 38525963