rollback

configuration.py

@@ -1,6 +1,6 @@
 class Config:
   num_frames = 8
-  frame_step = 5
+  frame_step = 15
   resolution = 224
   frame_size = (resolution, resolution)
   id_to_name = {
@@ -22,7 +22,7 @@ class Config:
 
   model_id = 'a0'
   detector_path = 'weights/yolov8n.pt'
-  classifier_path = 'weights/classifier-16-epoch9.keras'
+  classifier_path = 'weights/classifier-8-epoch10.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 model
+  # 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')
 
 def build_model(movinet, classifier_head):
   return keras.models.Sequential([movinet, classifier_head])

weights/classifier-8-epoch10.keras

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