.

.idea/workspace.xml

@@ -4,9 +4,10 @@
     <option name="autoReloadType" value="SELECTIVE" />
   </component>
   <component name="ChangeListManager">
-    <list default="true" id="5e4481c0-7ba2-42e4-bbe6-4c36a0d36baa" name="Changes" comment="font">
+    <list default="true" id="5e4481c0-7ba2-42e4-bbe6-4c36a0d36baa" name="Changes" comment="batch images">
       <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/CDM/detect_classify/classification.py" beforeDir="false" afterPath="$PROJECT_DIR$/CDM/detect_classify/classification.py" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/CDM/run_single.py" beforeDir="false" afterPath="$PROJECT_DIR$/CDM/run_single.py" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/app.py" beforeDir="false" afterPath="$PROJECT_DIR$/app.py" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/logs/app.log" beforeDir="false" afterPath="$PROJECT_DIR$/logs/app.log" afterDir="false" />
     </list>
@@ -104,14 +105,7 @@
       <workItem from="1723386970626" duration="451000" />
       <workItem from="1723387453009" duration="27328000" />
       <workItem from="1723978405562" duration="51637000" />
-      <workItem from="1725925225487" duration="136670000" />
-    </task>
-    <task id="LOCAL-00012" summary="google drive">
-      <created>1723456794848</created>
-      <option name="number" value="00012" />
-      <option name="presentableId" value="LOCAL-00012" />
-      <option name="project" value="LOCAL" />
-      <updated>1723456794848</updated>
+      <workItem from="1725925225487" duration="141465000" />
     </task>
     <task id="LOCAL-00013" summary="test auto delete">
       <created>1723458331923</created>
@@ -449,7 +443,14 @@
       <option name="project" value="LOCAL" />
       <updated>1728916518798</updated>
     </task>
-    <option name="localTasksCounter" value="61" />
+    <task id="LOCAL-00061" summary="batch images">
+      <created>1728919083991</created>
+      <option name="number" value="00061" />
+      <option name="presentableId" value="LOCAL-00061" />
+      <option name="project" value="LOCAL" />
+      <updated>1728919083992</updated>
+    </task>
+    <option name="localTasksCounter" value="62" />
     <servers />
   </component>
   <component name="TypeScriptGeneratedFilesManager">
@@ -487,9 +488,10 @@
     <MESSAGE value="." />
     <MESSAGE value="remove dotenv" />
     <MESSAGE value="font" />
-    <option name="LAST_COMMIT_MESSAGE" value="font" />
+    <MESSAGE value="batch images" />
+    <option name="LAST_COMMIT_MESSAGE" value="batch images" />
   </component>
   <component name="com.intellij.coverage.CoverageDataManagerImpl">
-    <SUITE FILE_PATH="coverage/Cpp4App_test$app.coverage" NAME="app Coverage Results" MODIFIED="1728918751288" 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/Cpp4App_test$app.coverage" NAME="app Coverage Results" MODIFIED="1728986730891" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="true" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
   </component>
 </project>

CDM/detect_classify/classification.py

@@ -177,7 +177,8 @@ def get_clf_model(clf_model="ResNet18", use_gpu=False):
 
     return model
 
-def compo_classification(input_img, output_root, segment_root, merge_json, output_data, resize_by_height=800, clf_model="ResNet18"):
+
+def compo_classification(input_img, output_root, segment_root, merge_json, output_data, resize_by_height=800, clf_model="ResNet18", model=get_clf_model("ResNet18")):
     # load text and non-text compo
     ele_id = 0
     compos = []
@@ -359,9 +360,10 @@ def compo_classification(input_img, output_root, segment_root, merge_json, outpu
 
 # =============================================================================
 
+    # model = get_clf_model(clf_model)
+
     classification_start_time = time.process_time()
 
-    model = get_clf_model(clf_model)
     comp_tensors = []
     elements = []
 

CDM/run_single.py

@@ -133,7 +133,7 @@ def run_single_img(input_img, output_root, segment_root):
         this_time_cost_ic, this_time_cost_ts, output_data, output_boards, classification_cost = clf.compo_classification(input_img, output_root,
                                                                                      segment_root, merge_json,
                                                                                      output_data,
-                                                                                     resize_by_height=resize_by_height, clf_model="ViT")
+                                                                                     resize_by_height=resize_by_height, clf_model="ViT", model = clf.get_clf_model("ViT"))
 
         ic_time_cost_all.append(this_time_cost_ic)
         ts_time_cost_all.append(this_time_cost_ts)

logs/app.log

@@ -5,3 +5,27 @@
 2024-10-15 02:12:40 - INFO - HTTP Request: GET https://checkip.amazonaws.com/ "HTTP/1.1 200 "
 2024-10-15 02:12:40 - INFO - HTTP Request: GET https://api.gradio.app/pkg-version "HTTP/1.1 200 OK"
 2024-10-15 02:12:40 - INFO - HTTP Request: GET https://checkip.amazonaws.com/ "HTTP/1.1 200 "
+2024-10-15 20:09:21 - INFO - Gradio app launched successfully
+2024-10-15 20:58:07 - INFO - Application started
+2024-10-15 20:58:07 - INFO - file_cache is only supported with oauth2client<4.0.0
+2024-10-15 20:58:07 - INFO - HTTP Request: GET http://127.0.0.1:7860/startup-events "HTTP/1.1 200 OK"
+2024-10-15 20:58:07 - INFO - HTTP Request: HEAD http://127.0.0.1:7860/ "HTTP/1.1 200 OK"
+2024-10-15 20:58:07 - INFO - HTTP Request: GET https://checkip.amazonaws.com/ "HTTP/1.1 200 "
+2024-10-15 20:58:07 - INFO - HTTP Request: GET https://checkip.amazonaws.com/ "HTTP/1.1 200 "
+2024-10-15 20:58:08 - INFO - HTTP Request: GET https://api.gradio.app/pkg-version "HTTP/1.1 200 OK"
+2024-10-15 20:58:55 - INFO - Gradio app launched successfully
+2024-10-15 20:59:04 - INFO - Application started
+2024-10-15 20:59:04 - INFO - file_cache is only supported with oauth2client<4.0.0
+2024-10-15 20:59:04 - INFO - HTTP Request: GET http://127.0.0.1:7860/startup-events "HTTP/1.1 200 OK"
+2024-10-15 20:59:04 - INFO - HTTP Request: HEAD http://127.0.0.1:7860/ "HTTP/1.1 200 OK"
+2024-10-15 20:59:04 - INFO - HTTP Request: GET https://checkip.amazonaws.com/ "HTTP/1.1 200 "
+2024-10-15 20:59:04 - INFO - HTTP Request: GET https://checkip.amazonaws.com/ "HTTP/1.1 200 "
+2024-10-15 20:59:04 - INFO - HTTP Request: GET https://api.gradio.app/pkg-version "HTTP/1.1 200 OK"
+2024-10-15 21:05:38 - INFO - Application started
+2024-10-15 21:05:38 - INFO - file_cache is only supported with oauth2client<4.0.0
+2024-10-15 21:05:38 - INFO - HTTP Request: GET http://127.0.0.1:7860/startup-events "HTTP/1.1 200 OK"
+2024-10-15 21:05:38 - INFO - HTTP Request: HEAD http://127.0.0.1:7860/ "HTTP/1.1 200 OK"
+2024-10-15 21:05:38 - INFO - HTTP Request: GET https://checkip.amazonaws.com/ "HTTP/1.1 200 "
+2024-10-15 21:05:38 - INFO - HTTP Request: GET https://checkip.amazonaws.com/ "HTTP/1.1 200 "
+2024-10-15 21:05:39 - INFO - HTTP Request: GET https://api.gradio.app/pkg-version "HTTP/1.1 200 OK"
+2024-10-15 21:54:10 - INFO - Gradio app launched successfully