yolo fire

.idea/workspace.xml

@@ -4,9 +4,12 @@
     <option name="autoReloadType" value="SELECTIVE" />
   </component>
   <component name="ChangeListManager">
-    <list default="true" id="d7806539-b6d6-42e7-bb45-1565f5d54891" name="Changes" comment="try to create /.cache folder">
+    <list default="true" id="d7806539-b6d6-42e7-bb45-1565f5d54891" name="Changes" comment="try to grant user can access `/.cache` folder">
+      <change afterPath="$PROJECT_DIR$/yolo_fire.py" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/Dockerfile" beforeDir="false" afterPath="$PROJECT_DIR$/Dockerfile" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/demo.py" beforeDir="false" afterPath="$PROJECT_DIR$/demo.py" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/requirements.txt" beforeDir="false" afterPath="$PROJECT_DIR$/requirements.txt" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -28,6 +31,9 @@
   <component name="MarkdownSettingsMigration">
     <option name="stateVersion" value="1" />
   </component>
+  <component name="ProblemsViewState">
+    <option name="selectedTabId" value="QODANA_PROBLEMS_VIEW_TAB" />
+  </component>
   <component name="ProjectColorInfo">{
   &quot;associatedIndex&quot;: 6
 }</component>
@@ -37,21 +43,22 @@
     <option name="hideEmptyMiddlePackages" value="true" />
     <option name="showLibraryContents" value="true" />
   </component>
-  <component name="PropertiesComponent"><![CDATA[{
-  "keyToString": {
-    "Python.demo.executor": "Run",
-    "RunOnceActivity.OpenProjectViewOnStart": "true",
-    "RunOnceActivity.ShowReadmeOnStart": "true",
-    "git-widget-placeholder": "main",
-    "last_opened_file_path": "/Users/famer.me/PycharmProjects/detector",
-    "node.js.detected.package.eslint": "true",
-    "node.js.detected.package.tslint": "true",
-    "node.js.selected.package.eslint": "(autodetect)",
-    "node.js.selected.package.tslint": "(autodetect)",
-    "nodejs_package_manager_path": "npm",
-    "vue.rearranger.settings.migration": "true"
+  <component name="PropertiesComponent">{
+  &quot;keyToString&quot;: {
+    &quot;Python.demo.executor&quot;: &quot;Run&quot;,
+    &quot;RunOnceActivity.OpenProjectViewOnStart&quot;: &quot;true&quot;,
+    &quot;RunOnceActivity.ShowReadmeOnStart&quot;: &quot;true&quot;,
+    &quot;git-widget-placeholder&quot;: &quot;main&quot;,
+    &quot;last_opened_file_path&quot;: &quot;/Users/famer.me/PycharmProjects/detector&quot;,
+    &quot;node.js.detected.package.eslint&quot;: &quot;true&quot;,
+    &quot;node.js.detected.package.tslint&quot;: &quot;true&quot;,
+    &quot;node.js.selected.package.eslint&quot;: &quot;(autodetect)&quot;,
+    &quot;node.js.selected.package.tslint&quot;: &quot;(autodetect)&quot;,
+    &quot;nodejs_package_manager_path&quot;: &quot;npm&quot;,
+    &quot;settings.editor.selected.configurable&quot;: &quot;settings.qodana&quot;,
+    &quot;vue.rearranger.settings.migration&quot;: &quot;true&quot;
   }
-}]]></component>
+}</component>
   <component name="RecentsManager">
     <key name="CopyFile.RECENT_KEYS">
       <recent name="$PROJECT_DIR$" />
@@ -118,7 +125,8 @@
       <option name="presentableId" value="Default" />
       <updated>1702301546100</updated>
       <workItem from="1702301554378" duration="4025000" />
-      <workItem from="1702450051021" duration="3998000" />
+      <workItem from="1702450051021" duration="10065000" />
+      <workItem from="1702476484538" duration="5222000" />
     </task>
     <task id="LOCAL-00001" summary="init commit">
       <option name="closed" value="true" />
@@ -160,18 +168,38 @@
       <option name="project" value="LOCAL" />
       <updated>1702453169015</updated>
     </task>
-    <option name="localTasksCounter" value="6" />
+    <task id="LOCAL-00006" summary="try to grant user can access `/.cache` folder">
+      <option name="closed" value="true" />
+      <created>1702454105421</created>
+      <option name="number" value="00006" />
+      <option name="presentableId" value="LOCAL-00006" />
+      <option name="project" value="LOCAL" />
+      <updated>1702454105421</updated>
+    </task>
+    <option name="localTasksCounter" value="7" />
     <servers />
   </component>
   <component name="TypeScriptGeneratedFilesManager">
     <option name="version" value="3" />
   </component>
+  <component name="Vcs.Log.Tabs.Properties">
+    <option name="TAB_STATES">
+      <map>
+        <entry key="MAIN">
+          <value>
+            <State />
+          </value>
+        </entry>
+      </map>
+    </option>
+  </component>
   <component name="VcsManagerConfiguration">
     <MESSAGE value="init commit" />
     <MESSAGE value="fix build" />
     <MESSAGE value="remove cv2" />
     <MESSAGE value="try to create /.cache folder" />
-    <option name="LAST_COMMIT_MESSAGE" value="try to create /.cache folder" />
+    <MESSAGE value="try to grant user can access `/.cache` folder" />
+    <option name="LAST_COMMIT_MESSAGE" value="try to grant user can access `/.cache` folder" />
   </component>
   <component name="com.intellij.coverage.CoverageDataManagerImpl">
     <SUITE FILE_PATH="coverage/detector$demo.coverage" NAME="demo Coverage Results" MODIFIED="1702302380580" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="true" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />

Dockerfile

@@ -11,4 +11,5 @@ RUN mkdir $HOME/.cache
 RUN chown -R user $HOME/.cache
 
 RUN pip install -r requirements.txt
+RUN python yolo_fire.py
 ENTRYPOINT ["streamlit", "run", "demo.py"]

demo.py

@@ -3,7 +3,7 @@ import time
 from PIL import Image
 import streamlit as st
 
-from model import detect
+from yolo_fire import detect
 
 
 def main():

requirements.txt

@@ -5,64 +5,90 @@ attrs==23.1.0
 blinker==1.7.0
 cachetools==5.3.2
 certifi==2023.11.17
+chardet==4.0.0
 charset-normalizer==3.3.2
 click==8.1.7
+contourpy==1.2.0
+cycler==0.10.0
 exceptiongroup==1.2.0
 fastapi==0.104.1
 filelock==3.13.1
+fire==0.5.0
+fonttools==4.46.0
 fsspec==2023.12.1
 gitdb==4.0.11
 GitPython==3.1.40
 h11==0.14.0
 httptools==0.6.1
+huggingface-hub==0.19.4
 idna==3.6
 importlib-metadata==6.11.0
+importlib-resources==6.1.1
 Jinja2==3.1.2
 jsonschema==4.20.0
 jsonschema-specifications==2023.11.2
+kiwisolver==1.4.5
 markdown-it-py==3.0.0
 MarkupSafe==2.1.3
+matplotlib==3.8.2
 mdurl==0.1.2
 mpmath==1.3.0
 networkx==3.2.1
 numpy==1.26.2
+opencv-python==4.7.0.72
+opencv-python-headless==4.8.0.74
 packaging==23.2
 pandas==2.1.4
 Pillow==10.1.0
 protobuf==4.25.1
+psutil==5.9.6
+py-cpuinfo==9.0.0
 pyarrow==14.0.1
+pybboxes==0.1.6
 pydantic==2.5.2
 pydantic_core==2.14.5
 pydeck==0.8.1b0
 Pygments==2.17.2
+pyparsing==2.4.7
 python-dateutil==2.8.2
 python-dotenv==1.0.0
+python-magic==0.4.27
 pytz==2023.3.post1
 PyYAML==6.0.1
 referencing==0.32.0
 requests==2.31.0
+requests-toolbelt==1.0.0
 rich==13.7.0
+roboflow==1.1.12
 rpds-py==0.13.2
+sahi==0.11.15
 scipy==1.11.4
+seaborn==0.13.0
+shapely==2.0.2
 six==1.16.0
 smmap==5.0.1
 sniffio==1.3.0
 starlette==0.27.0
 streamlit==1.29.0
+supervision==0.17.1
 sympy==1.12
 tenacity==8.2.3
+termcolor==2.4.0
+terminaltables==3.1.10
+thop==0.1.1.post2209072238
 toml==0.10.2
 toolz==0.12.0
 torch==2.1.1
 torchaudio==2.1.1
 torchvision==0.16.1
 tornado==6.4
+tqdm==4.66.1
 typing_extensions==4.9.0
 tzdata==2023.3
 tzlocal==5.2
+ultralytics==8.0.227
 urllib3==2.1.0
 uvicorn==0.24.0.post1
-uvloop==0.19.0
 validators==0.22.0
 watchfiles==0.21.0
 websockets==12.0

yolo_fire.py

@@ -0,0 +1,51 @@
+import random
+from PIL import ImageDraw, Image
+from huggingface_hub import hf_hub_download
+from ultralytics import YOLO
+
+# Load model
+weight_file = hf_hub_download("SHOU-ISD/fire-and-smoke", "yolov8n.pt")
+model = YOLO(weight_file)  # pretrained YOLOv8n model
+
+# Helper Functions for Plotting BBoxes
+def plot_one_box(x, img, color=None, label=None, line_thickness=None):
+    width, height = img.size
+    tl = line_thickness or round(0.002 * (width + height) / 2) + 1  # line/font thickness
+    color = color or (random.randint(0, 255), random.randint(0, 255), random.randint(0, 255))
+    c1, c2 = (int(x[0]), int(x[1])), (int(x[2]), int(x[3]))
+    img_draw = ImageDraw.Draw(img)
+    img_draw.rectangle((c1[0], c1[1], c2[0], c2[1]), outline=color, width=tl)
+    if label:
+        tf = max(tl - 1, 1)  # font thickness
+        x1, y1, x2, y2 = img_draw.textbbox(c1, label, stroke_width=tf)
+        img_draw.rectangle((x1, y1, x2, y2), fill=color)
+        img_draw.text((x1, y1), label, fill=(255, 255, 255))
+
+
+# Ploting Bounding Box on img
+def add_bboxes(pil_img, result, confidence=0.6):
+    for box in result.boxes:
+        [cl] = box.cls.tolist()
+        [conf] = box.conf.tolist()
+        if conf < confidence:
+            continue
+        [rect] = box.xyxy.tolist()
+        text = f'{result.names[cl]}: {conf: 0.2f}'
+        plot_one_box(x=rect, img=pil_img, label=text)
+
+    return pil_img
+
+
+def detect(im, confidence):
+    results = model(source=im)
+    res_img = im
+    for result in results:
+        res_img = add_bboxes(res_img, result, confidence)
+    return res_img
+
+if __name__ == '__main__':
+    im = Image.open("./tests/fire1.jpg")
+    results = model(source=im)
+    for result in results:
+        im = add_bboxes(im, result)
+    im.show()