diff --git a/classifier/configs/phase4/_base.json b/classifier/configs/phase4/_base.json new file mode 100644 index 0000000..5e3eb83 --- /dev/null +++ b/classifier/configs/phase4/_base.json @@ -0,0 +1,7 @@ +{ + "pretrained": true, + "epochs": 15, + "image_size": 224, + "augment": false, + "data_dir": "cropped/classifier" +} diff --git a/classifier/configs/phase4/p4a_convnext_tiny_100pct.json b/classifier/configs/phase4/p4a_convnext_tiny_100pct.json new file mode 100644 index 0000000..e841de4 --- /dev/null +++ b/classifier/configs/phase4/p4a_convnext_tiny_100pct.json @@ -0,0 +1,6 @@ +{ + "extends": "_base.json", + "run_name": "p4a_convnext_tiny_100pct", + "backbone": "convnext_tiny", + "subsample": 1.0 +} diff --git a/classifier/configs/phase4/p4a_convnext_tiny_50pct.json b/classifier/configs/phase4/p4a_convnext_tiny_50pct.json new file mode 100644 index 0000000..f716886 --- /dev/null +++ b/classifier/configs/phase4/p4a_convnext_tiny_50pct.json @@ -0,0 +1,6 @@ +{ + "extends": "_base.json", + "run_name": "p4a_convnext_tiny_50pct", + "backbone": "convnext_tiny", + "subsample": 0.5 +} diff --git a/classifier/configs/phase4/p4a_efficientnet_b0_100pct.json b/classifier/configs/phase4/p4a_efficientnet_b0_100pct.json new file mode 100644 index 0000000..5a5c3b1 --- /dev/null +++ b/classifier/configs/phase4/p4a_efficientnet_b0_100pct.json @@ -0,0 +1,6 @@ +{ + "extends": "_base.json", + "run_name": "p4a_efficientnet_b0_100pct", + "backbone": "efficientnet_b0", + "subsample": 1.0 +} diff --git a/classifier/configs/phase4/p4a_efficientnet_b0_50pct.json b/classifier/configs/phase4/p4a_efficientnet_b0_50pct.json new file mode 100644 index 0000000..214daf4 --- /dev/null +++ b/classifier/configs/phase4/p4a_efficientnet_b0_50pct.json @@ -0,0 +1,6 @@ +{ + "extends": "_base.json", + "run_name": "p4a_efficientnet_b0_50pct", + "backbone": "efficientnet_b0", + "subsample": 0.5 +} diff --git a/classifier/configs/phase4/p4a_resnet50_100pct.json b/classifier/configs/phase4/p4a_resnet50_100pct.json new file mode 100644 index 0000000..b4824f1 --- /dev/null +++ b/classifier/configs/phase4/p4a_resnet50_100pct.json @@ -0,0 +1,6 @@ +{ + "extends": "_base.json", + "run_name": "p4a_resnet50_100pct", + "backbone": "resnet50", + "subsample": 1.0 +} diff --git a/classifier/configs/phase4/p4a_resnet50_50pct.json b/classifier/configs/phase4/p4a_resnet50_50pct.json new file mode 100644 index 0000000..7690f47 --- /dev/null +++ b/classifier/configs/phase4/p4a_resnet50_50pct.json @@ -0,0 +1,6 @@ +{ + "extends": "_base.json", + "run_name": "p4a_resnet50_50pct", + "backbone": "resnet50", + "subsample": 0.5 +}