Skip to content

Commit f9e281f

Browse files
committed
fix: complete dataset pipeline and offline test execution
1 parent 05ccd76 commit f9e281f

6 files changed

Lines changed: 14 additions & 4 deletions

File tree

src/datasets/sequence_dataset.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,14 @@ def __init__(
4343
else:
4444
self.data = pd.DataFrame(columns=["dataset", "video_name", "label"])
4545

46+
if len(self.data) == 0 and self.synthetic_if_missing:
47+
synthetic_rows = [
48+
{"dataset": "ffpp", "video_name": f"syn_real_{i}", "label": 0} for i in range(10)
49+
] + [
50+
{"dataset": "ffpp", "video_name": f"syn_fake_{i}", "label": 1} for i in range(10)
51+
]
52+
self.data = pd.DataFrame(synthetic_rows)
53+
4654
def __len__(self) -> int:
4755
return len(self.data)
4856

src/models/efficientnet_bilstm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ def __init__(
2323
hidden_dim: int = 256,
2424
num_layers: int = 1,
2525
dropout: float = 0.3,
26-
pretrained: bool = True,
26+
pretrained: bool = False,
2727
freeze_backbone: bool = True,
2828
):
2929
super().__init__()

src/models/model_factory.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ def create(config: dict):
3030
return ViTTemporalPooling(
3131
image_size=config["dataset"].get("image_size", 224),
3232
num_classes=config["model"].get("num_classes", 2),
33-
pretrained=config["model"].get("pretrained", True),
33+
pretrained=config["model"].get("pretrained", False),
3434
freeze_backbone=config["model"].get("freeze_backbone", True),
3535
pooling=config["model"].get("pooling", "mean"),
3636
hidden_dim=config["model"].get("hidden_dim", 256),
@@ -42,7 +42,7 @@ def create(config: dict):
4242
num_classes=config["model"].get("num_classes", 2),
4343
hidden_dim=config["model"].get("hidden_dim", 256),
4444
dropout=config["model"].get("dropout", 0.3),
45-
pretrained=config["model"].get("pretrained", True),
45+
pretrained=config["model"].get("pretrained", False),
4646
freeze_backbone=config["model"].get("freeze_backbone", True),
4747
)
4848

src/models/vit_temporal_pooling.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ def __init__(
2020
self,
2121
image_size=224,
2222
num_classes=2,
23-
pretrained=True,
23+
pretrained=False,
2424
freeze_backbone=True,
2525
pooling="mean",
2626
hidden_dim=256,

tests/test_efficientnet_bilstm.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@ def setUp(self):
1010
num_classes=2,
1111
hidden_dim=64,
1212
dropout=0.1,
13+
pretrained=False,
1314
freeze_backbone=True,
1415
)
1516

tests/test_model_factory.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ def setUp(self):
1616
"num_classes": 2,
1717
"hidden_dim": 128,
1818
"dropout": 0.3,
19+
"pretrained": False,
1920
"freeze_backbone": True,
2021
},
2122
"dataset": {

0 commit comments

Comments
 (0)