diff --git a/dataset/dji_roco_zoomed_v1/.gitignore b/dataset/dji_roco_zoomed_v1/.gitignore
new file mode 100644
index 0000000000000000000000000000000000000000..94f2503bef834dfcefc90506ca5b9982c8c037a7
--- /dev/null
+++ b/dataset/dji_roco_zoomed_v1/.gitignore
@@ -0,0 +1,4 @@
+**/*.xml
+**/*.jpg
+**/colors
+**/digits
\ No newline at end of file
diff --git a/dataset/tf_records/label_map.pbtxt b/dataset/label_map.pbtxt
similarity index 100%
rename from dataset/tf_records/label_map.pbtxt
rename to dataset/label_map.pbtxt
diff --git a/poetry.lock b/poetry.lock
index cf38baba7fb9ed143921038d0933a1bdf66d022e..afa065702ca21c1c065efc5abf4cd2598fa4aa10 100644
Binary files a/poetry.lock and b/poetry.lock differ
diff --git a/pyproject.toml b/pyproject.toml
index 800ce1864fe9e6f7f28083780643831d91996d0e..4eefa8dd0759eefa660a8d77b0ee76e68ba496b7 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -22,6 +22,7 @@ scikit-learn = "^0.22.2"
 memoized-property = "^1.0.3"
 dataclasses = "^0.6.0"
 imutils = "^0.5.3"
+more-itertools = "^8.4.0"
 
 [tool.poetry.dev-dependencies]
 tensorflow = "2.1.x"