Skip to content

affective_computing / face_mask_classifier

Go to a project
Toggle navigation
Toggle navigation pinning
801e5009 by 乔峰昇
Options

mnn

1 parent 4d8ada78
Inline Side-by-side
Showing 9 changed files with 80 additions and 24 deletions
.idea/.gitignore 0 → 100644
1 # Default ignored files
2 /shelf/
3 /workspace.xml
.idea/face_mask_classifier.iml 0 → 100644
1 <?xml version="1.0" encoding="UTF-8"?>
2 <module type="PYTHON_MODULE" version="4">
3 <component name="NewModuleRootManager">
4 <content url="file://$MODULE_DIR$" />
5 <orderEntry type="jdk" jdkName="Python 3.8 (gan)" jdkType="Python SDK" />
6 <orderEntry type="sourceFolder" forTests="false" />
7 </component>
8 </module>
...\ No newline at end of file ...\ No newline at end of file
.idea/inspectionProfiles/Project_Default.xml 0 → 100644
1 <component name="InspectionProjectProfileManager">
2 <profile version="1.0">
3 <option name="myName" value="Project Default" />
4 <inspection_tool class="PyPackageRequirementsInspection" enabled="true" level="WARNING" enabled_by_default="true">
5 <option name="ignoredPackages">
6 <value>
7 <list size="25">
8 <item index="0" class="java.lang.String" itemvalue="tqdm" />
9 <item index="1" class="java.lang.String" itemvalue="easydict" />
10 <item index="2" class="java.lang.String" itemvalue="scikit_image" />
11 <item index="3" class="java.lang.String" itemvalue="matplotlib" />
12 <item index="4" class="java.lang.String" itemvalue="tensorboardX" />
13 <item index="5" class="java.lang.String" itemvalue="torch" />
14 <item index="6" class="java.lang.String" itemvalue="numpy" />
15 <item index="7" class="java.lang.String" itemvalue="pycocotools" />
16 <item index="8" class="java.lang.String" itemvalue="skimage" />
17 <item index="9" class="java.lang.String" itemvalue="Pillow" />
18 <item index="10" class="java.lang.String" itemvalue="scipy" />
19 <item index="11" class="java.lang.String" itemvalue="torchvision" />
20 <item index="12" class="java.lang.String" itemvalue="opencv_python" />
21 <item index="13" class="java.lang.String" itemvalue="onnxruntime" />
22 <item index="14" class="java.lang.String" itemvalue="onnx-simplifier" />
23 <item index="15" class="java.lang.String" itemvalue="onnx" />
24 <item index="16" class="java.lang.String" itemvalue="opencv-contrib-python" />
25 <item index="17" class="java.lang.String" itemvalue="numba" />
26 <item index="18" class="java.lang.String" itemvalue="opencv-python" />
27 <item index="19" class="java.lang.String" itemvalue="librosa" />
28 <item index="20" class="java.lang.String" itemvalue="tensorboard" />
29 <item index="21" class="java.lang.String" itemvalue="dill" />
30 <item index="22" class="java.lang.String" itemvalue="pandas" />
31 <item index="23" class="java.lang.String" itemvalue="scikit_learn" />
32 <item index="24" class="java.lang.String" itemvalue="pytorch-gradual-warmup-lr" />
33 </list>
34 </value>
35 </option>
36 </inspection_tool>
37 </profile>
38 </component>
...\ No newline at end of file ...\ No newline at end of file
.idea/inspectionProfiles/profiles_settings.xml 0 → 100644
1 <component name="InspectionProjectProfileManager">
2 <settings>
3 <option name="USE_PROJECT_PROFILE" value="false" />
4 <version value="1.0" />
5 </settings>
6 </component>
...\ No newline at end of file ...\ No newline at end of file
.idea/misc.xml 0 → 100644
1 <?xml version="1.0" encoding="UTF-8"?>
2 <project version="4">
3 <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.8 (gan)" project-jdk-type="Python SDK" />
4 </project>
...\ No newline at end of file ...\ No newline at end of file
.idea/modules.xml 0 → 100644
1 <?xml version="1.0" encoding="UTF-8"?>
2 <project version="4">
3 <component name="ProjectModuleManager">
4 <modules>
5 <module fileurl="file://$PROJECT_DIR$/.idea/face_mask_classifier.iml" filepath="$PROJECT_DIR$/.idea/face_mask_classifier.iml" />
6 </modules>
7 </component>
8 </project>
...\ No newline at end of file ...\ No newline at end of file
.idea/vcs.xml 0 → 100644
1 <?xml version="1.0" encoding="UTF-8"?>
2 <project version="4">
3 <component name="VcsDirectoryMappings">
4 <mapping directory="$PROJECT_DIR$" vcs="Git" />
5 </component>
6 </project>
...\ No newline at end of file ...\ No newline at end of file
.idea/workspace.xml
1 <?xml version="1.0" encoding="UTF-8"?> 1 <?xml version="1.0" encoding="UTF-8"?>
2 <project version="4"> 2 <project version="4">
3 <component name="ChangeListManager"> 3 <component name="ChangeListManager">
4 <list default="true" id="a64a78af-ad8c-4647-b359-632e9aeec7f0" name="Default Changelist" comment=""> 4 <list default="true" id="e3705fa5-39c9-4348-8902-c51f78f5999f" name="Default Changelist" comment="">
5 <change beforePath="$PROJECT_DIR$/.idea/face_mask_classifier.iml" beforeDir="false" /> 5 <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
6 <change beforePath="$PROJECT_DIR$/.idea/inspectionProfiles/Project_Default.xml" beforeDir="false" />
7 <change beforePath="$PROJECT_DIR$/.idea/inspectionProfiles/profiles_settings.xml" beforeDir="false" />
8 <change beforePath="$PROJECT_DIR$/.idea/misc.xml" beforeDir="false" />
9 <change beforePath="$PROJECT_DIR$/.idea/modules.xml" beforeDir="false" />
10 <change beforePath="$PROJECT_DIR$/.idea/vcs.xml" beforeDir="false" />
11 <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" />
12 <change beforePath="$PROJECT_DIR$/cls_abnormal_face_mnn_1.0.0_v0.0.1.mnn" beforeDir="false" afterPath="$PROJECT_DIR$/cls_abnormal_face_mnn_1.0.0_v0.0.1.mnn" afterDir="false" />
13 <change beforePath="$PROJECT_DIR$/cls_abnormal_face_onnx_1.0.0_v0.0.1.onnx" beforeDir="false" />
14 <change beforePath="$PROJECT_DIR$/infer_mnn.py" beforeDir="false" afterPath="$PROJECT_DIR$/infer_mnn.py" afterDir="false" /> 6 <change beforePath="$PROJECT_DIR$/infer_mnn.py" beforeDir="false" afterPath="$PROJECT_DIR$/infer_mnn.py" afterDir="false" />
15 <change beforePath="$PROJECT_DIR$/mobilenet_v2/logs/events.out.tfevents.1644369407.USER-20210707NI.5140.0" beforeDir="false" />
16 <change beforePath="$PROJECT_DIR$/mobilenet_v2/weight/best.pth" beforeDir="false" afterPath="$PROJECT_DIR$/mobilenet_v2/weight/best.pth" afterDir="false" />
17 <change beforePath="$PROJECT_DIR$/mobilenet_v2/weight/last.pth" beforeDir="false" afterPath="$PROJECT_DIR$/mobilenet_v2/weight/last.pth" afterDir="false" />
18 <change beforePath="$PROJECT_DIR$/model/dataset/dataset.py" beforeDir="false" afterPath="$PROJECT_DIR$/model/dataset/dataset.py" afterDir="false" />
19 <change beforePath="$PROJECT_DIR$/model/utils/utils.py" beforeDir="false" afterPath="$PROJECT_DIR$/model/utils/utils.py" afterDir="false" />
20 </list> 7 </list>
21 <option name="SHOW_DIALOG" value="false" /> 8 <option name="SHOW_DIALOG" value="false" />
22 <option name="HIGHLIGHT_CONFLICTS" value="true" /> 9 <option name="HIGHLIGHT_CONFLICTS" value="true" />
...@@ -80,15 +67,15 @@ ...@@ -80,15 +67,15 @@
80 <method v="2" /> 67 <method v="2" />
81 </configuration> 68 </configuration>
82 <configuration name="infer_mnn" type="PythonConfigurationType" factoryName="Python" temporary="true" nameIsGenerated="true"> 69 <configuration name="infer_mnn" type="PythonConfigurationType" factoryName="Python" temporary="true" nameIsGenerated="true">
83 <module name="pytorch-image-classifier-collection" /> 70 <module name="face_mask_classifier" />
84 <option name="INTERPRETER_OPTIONS" value="" /> 71 <option name="INTERPRETER_OPTIONS" value="" />
85 <option name="PARENT_ENVS" value="true" /> 72 <option name="PARENT_ENVS" value="true" />
86 <envs> 73 <envs>
87 <env name="PYTHONUNBUFFERED" value="1" /> 74 <env name="PYTHONUNBUFFERED" value="1" />
88 </envs> 75 </envs>
89 <option name="SDK_HOME" value="" /> 76 <option name="SDK_HOME" value="E:\ProgramData\Anaconda3\envs\gan\python.exe" />
90 <option name="WORKING_DIRECTORY" value="$PROJECT_DIR$" /> 77 <option name="WORKING_DIRECTORY" value="$PROJECT_DIR$" />
91 <option name="IS_MODULE_SDK" value="true" /> 78 <option name="IS_MODULE_SDK" value="false" />
92 <option name="ADD_CONTENT_ROOTS" value="true" /> 79 <option name="ADD_CONTENT_ROOTS" value="true" />
93 <option name="ADD_SOURCE_ROOTS" value="true" /> 80 <option name="ADD_SOURCE_ROOTS" value="true" />
94 <option name="SCRIPT_NAME" value="$PROJECT_DIR$/infer_mnn.py" /> 81 <option name="SCRIPT_NAME" value="$PROJECT_DIR$/infer_mnn.py" />
...@@ -154,7 +141,7 @@ ...@@ -154,7 +141,7 @@
154 <component name="SpellCheckerSettings" RuntimeDictionaries="0" Folders="0" CustomDictionaries="0" DefaultDictionary="application-level" UseSingleDictionary="true" transferred="true" /> 141 <component name="SpellCheckerSettings" RuntimeDictionaries="0" Folders="0" CustomDictionaries="0" DefaultDictionary="application-level" UseSingleDictionary="true" transferred="true" />
155 <component name="TaskManager"> 142 <component name="TaskManager">
156 <task active="true" id="Default" summary="Default task"> 143 <task active="true" id="Default" summary="Default task">
157 <changelist id="a64a78af-ad8c-4647-b359-632e9aeec7f0" name="Default Changelist" comment="" /> 144 <changelist id="e3705fa5-39c9-4348-8902-c51f78f5999f" name="Default Changelist" comment="" />
158 <created>1644369279294</created> 145 <created>1644369279294</created>
159 <option name="number" value="Default" /> 146 <option name="number" value="Default" />
160 <option name="presentableId" value="Default" /> 147 <option name="presentableId" value="Default" />
......
infer_mnn.py
1 import os
2
3 import cv2 1 import cv2
4 from PIL import Image, ImageFont, ImageDraw
5 import numpy as np 2 import numpy as np
6 from torchvision import transforms
7 import MNN 3 import MNN
8 4 import os
9 5
10 def image_infer_mnn(mnn_model_path, image_path, class_list): 6 def image_infer_mnn(mnn_model_path, image_path, class_list):
11 image = cv2.imread(image_path) 7 image = cv2.imread(image_path)
......
Styling with Markdown is supported
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!