From 5bdf860786d4327a24c8fb562f34343477f0df2e Mon Sep 17 00:00:00 2001
From: Mathieu Beligon <mathieu@feedly.com>
Date: Sat, 4 Jul 2020 22:05:12 +0200
Subject: [PATCH] [common] (records) factorize code
---
.../common/scripts/create_tensorflow_records.py | 16 ++++++++++++----
1 file changed, 12 insertions(+), 4 deletions(-)
diff --git a/common/research/common/scripts/create_tensorflow_records.py b/common/research/common/scripts/create_tensorflow_records.py
index 9227561..9ca8456 100644
--- a/common/research/common/scripts/create_tensorflow_records.py
+++ b/common/research/common/scripts/create_tensorflow_records.py
@@ -12,9 +12,7 @@ def create_one_record_per_roco_dset():
TensorflowRecordFactory.from_dataset(roco_set)
-if __name__ == "__main__":
- # create_one_record_per_roco_dset()
-
+def create_twitch_records():
TensorflowRecordFactory.from_dataset(
UnionDataset(
TwitchROCODataset.TWITCH_470149568,
@@ -24,21 +22,31 @@ if __name__ == "__main__":
TwitchROCODataset.TWITCH_470152730,
)
)
-
TensorflowRecordFactory.from_dataset(
UnionDataset(
TwitchROCODataset.TWITCH_470152838, TwitchROCODataset.TWITCH_470153081, TwitchROCODataset.TWITCH_470158483,
)
)
+
+def create_dji_records():
TensorflowRecordFactory.from_dataset(
UnionDataset(DJIROCODataset.CentralChina, DJIROCODataset.NorthChina, DJIROCODataset.SouthChina)
)
TensorflowRecordFactory.from_dataset(DJIROCODataset.Final)
+
+def create_dji_zoomed_records():
TensorflowRecordFactory.from_dataset(
UnionDataset(
DJIROCOZoomedDataset.CentralChina, DJIROCOZoomedDataset.NorthChina, DJIROCOZoomedDataset.SouthChina
)
)
TensorflowRecordFactory.from_dataset(DJIROCOZoomedDataset.Final)
+
+
+if __name__ == "__main__":
+ create_one_record_per_roco_dset()
+ create_twitch_records()
+ create_dji_records()
+ create_dji_zoomed_records()
--
GitLab