1 parent 05ccd76 commit f9e281fCopy full SHA for f9e281f
6 files changed
src/datasets/sequence_dataset.py
@@ -43,6 +43,14 @@ def __init__(
43
else:
44
self.data = pd.DataFrame(columns=["dataset", "video_name", "label"])
45
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
+
54
def __len__(self) -> int:
55
return len(self.data)
56
src/models/efficientnet_bilstm.py
@@ -23,7 +23,7 @@ def __init__(
23
hidden_dim: int = 256,
24
num_layers: int = 1,
25
dropout: float = 0.3,
26
- pretrained: bool = True,
+ pretrained: bool = False,
27
freeze_backbone: bool = True,
28
):
29
super().__init__()
src/models/model_factory.py
@@ -30,7 +30,7 @@ def create(config: dict):
30
return ViTTemporalPooling(
31
image_size=config["dataset"].get("image_size", 224),
32
num_classes=config["model"].get("num_classes", 2),
33
- pretrained=config["model"].get("pretrained", True),
+ pretrained=config["model"].get("pretrained", False),
34
freeze_backbone=config["model"].get("freeze_backbone", True),
35
pooling=config["model"].get("pooling", "mean"),
36
hidden_dim=config["model"].get("hidden_dim", 256),
@@ -42,7 +42,7 @@ def create(config: dict):
42
dropout=config["model"].get("dropout", 0.3),
)
src/models/vit_temporal_pooling.py
@@ -20,7 +20,7 @@ def __init__(
20
self,
21
image_size=224,
22
num_classes=2,
- pretrained=True,
+ pretrained=False,
freeze_backbone=True,
pooling="mean",
hidden_dim=256,
tests/test_efficientnet_bilstm.py
@@ -10,6 +10,7 @@ def setUp(self):
10
11
hidden_dim=64,
12
dropout=0.1,
13
14
15
16
tests/test_model_factory.py
@@ -16,6 +16,7 @@ def setUp(self):
"num_classes": 2,
17
"hidden_dim": 128,
18
"dropout": 0.3,
19
+ "pretrained": False,
"freeze_backbone": True,
},
"dataset": {
0 commit comments