filter out macos cache files

.idea/workspace.xml

@@ -6,6 +6,7 @@
   <component name="ChangeListManager">
     <list default="true" id="d7806539-b6d6-42e7-bb45-1565f5d54891" name="Changes" comment="fix runtime error in coco evaluator">
       <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/demo.py" beforeDir="false" afterPath="$PROJECT_DIR$/demo.py" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/yolo_dataset.py" beforeDir="false" afterPath="$PROJECT_DIR$/yolo_dataset.py" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
@@ -238,7 +239,7 @@
       <workItem from="1706274709532" duration="27000" />
       <workItem from="1706445222599" duration="1814000" />
       <workItem from="1706449264770" duration="4974000" />
-      <workItem from="1706517295469" duration="574000" />
+      <workItem from="1706517295469" duration="1726000" />
     </task>
     <task id="LOCAL-00001" summary="init commit">
       <option name="closed" value="true" />
@@ -464,7 +465,15 @@
       <option name="project" value="LOCAL" />
       <updated>1706454211376</updated>
     </task>
-    <option name="localTasksCounter" value="29" />
+    <task id="LOCAL-00029" summary="fix runtime error in coco evaluator">
+      <option name="closed" value="true" />
+      <created>1706517885858</created>
+      <option name="number" value="00029" />
+      <option name="presentableId" value="LOCAL-00029" />
+      <option name="project" value="LOCAL" />
+      <updated>1706517885858</updated>
+    </task>
+    <option name="localTasksCounter" value="30" />
     <servers />
   </component>
   <component name="TypeScriptGeneratedFilesManager">
@@ -519,6 +528,16 @@
           <line>77</line>
           <option name="timeStamp" value="17" />
         </line-breakpoint>
+        <line-breakpoint enabled="true" suspend="THREAD" type="python-line">
+          <url>file://$PROJECT_DIR$/yolo_dataset.py</url>
+          <line>239</line>
+          <option name="timeStamp" value="20" />
+        </line-breakpoint>
+        <line-breakpoint enabled="true" suspend="THREAD" type="python-line">
+          <url>file://$PROJECT_DIR$/yolo_dataset.py</url>
+          <line>229</line>
+          <option name="timeStamp" value="21" />
+        </line-breakpoint>
       </breakpoints>
     </breakpoint-manager>
   </component>
@@ -526,7 +545,7 @@
     <SUITE FILE_PATH="coverage/detector$yolo_dataset.coverage" NAME="yolo_dataset Coverage Results" MODIFIED="1705852113469" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="true" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
     <SUITE FILE_PATH="coverage/detector$evaluator.coverage" NAME="evaluator Coverage Results" MODIFIED="1706107083258" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="true" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
     <SUITE FILE_PATH="coverage/detector$demo.coverage" NAME="demo Coverage Results" MODIFIED="1706108414052" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="true" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
-    <SUITE FILE_PATH="coverage/detector$streamlit.coverage" NAME="streamlit Coverage Results" MODIFIED="1706517471264" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="true" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="" />
+    <SUITE FILE_PATH="coverage/detector$streamlit.coverage" NAME="streamlit Coverage Results" MODIFIED="1706518941359" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="true" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="" />
     <SUITE FILE_PATH="coverage/detector$extract.coverage" NAME="yolo_dataset Coverage Results" MODIFIED="1705764465837" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="true" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
   </component>
 </project>

demo.py

@@ -25,7 +25,7 @@ def main():
 
     model = None
     with st.sidebar:
-        model_choice = st.radio("Select Model", ["Fire&Smoke", "Crack", "Coco"])
+        model_choice = st.radio("Select Model", ["Fire&Smoke", "Crack"])
         if model_choice == "Fire&Smoke":
             model = fire_and_smoke
         elif model_choice == "Crack":

yolo_dataset.py

@@ -127,7 +127,7 @@ class YoloDataset:
     @staticmethod
     def from_zip_file(zip_file: ZipFile) -> 'YoloDataset':
         namelist = zip_file.namelist()
-        namelist = list(filter(lambda x: not x.startswith('.DS_Store'), namelist))
+        namelist = list(filter(lambda x: not (x.endswith('.DS_Store') or x.startswith('__MACOSX')), namelist))
         root_name = namelist[0]
         if not zip_file.getinfo(root_name).is_dir():
             root_name = root_name.split('/')[0] + '/'
@@ -228,7 +228,7 @@ class MaterialYoloDataset:
         self._zip_file = dataset.zip_file
         first = self._zip_file.namelist()[0]
         if self._zip_file.getinfo(first).is_dir():
-            self._root = first[-1]
+            self._root = first[0:-1]
         else:
             self._root = first.split('/')[0]
 
@@ -239,8 +239,8 @@ class MaterialYoloDataset:
         self._zip_file.extractall(f'./datasets/')
         with open(f'./datasets/{self._root}/data.yaml', 'w+') as f:
             f.write(f'path: {dataset_path.absolute()}/{self._root}/\n')
-            f.write(f'train: images/train\n')
-            f.write(f'val: images/val\n')
+            f.write(f'train: images\n')
+            f.write(f'val: images\n')
             f.write(f'\n')
             f.write(f'# Classes\n')
             f.write(f'names:\n')