diff --git a/.gitattributes b/.gitattributes index e92cf9b..7a34f23 100644 --- a/.gitattributes +++ b/.gitattributes @@ -1 +1,2 @@ *.pt filter=lfs diff=lfs merge=lfs -text +*.pdiparams filter=lfs diff=lfs merge=lfs -text diff --git a/.gitignore b/.gitignore index 855dbb4..e201a6a 100644 --- a/.gitignore +++ b/.gitignore @@ -73,5 +73,4 @@ behavior_detection/ fire_detection/ safety/ yolov/ -models/ __pycache__/ diff --git a/models/smoking_detection_paddle/infer_cfg.yml b/models/smoking_detection_paddle/infer_cfg.yml new file mode 100644 index 0000000..77d4ac3 --- /dev/null +++ b/models/smoking_detection_paddle/infer_cfg.yml @@ -0,0 +1,26 @@ +mode: paddle +draw_threshold: 0.5 +metric: COCO +use_dynamic_shape: false +arch: YOLO +min_subgraph_size: 3 +Preprocess: +- interp: 2 + keep_ratio: false + target_size: + - 640 + - 640 + type: Resize +- is_scale: true + mean: + - 0.485 + - 0.456 + - 0.406 + std: + - 0.229 + - 0.224 + - 0.225 + type: NormalizeImage +- type: Permute +label_list: +- cigarette diff --git a/models/smoking_detection_paddle/infer_cfg_backup.yml b/models/smoking_detection_paddle/infer_cfg_backup.yml new file mode 100644 index 0000000..9a1db44 --- /dev/null +++ b/models/smoking_detection_paddle/infer_cfg_backup.yml @@ -0,0 +1,105 @@ +mode: paddle +draw_threshold: 0.5 +metric: COCO +use_dynamic_shape: false +arch: YOLO +min_subgraph_size: 3 +Preprocess: +- interp: 2 + keep_ratio: false + target_size: + - 640 + - 640 + type: Resize +- is_scale: true + mean: + - 0.485 + - 0.456 + - 0.406 + std: + - 0.229 + - 0.224 + - 0.225 + type: NormalizeImage +- type: Permute +label_list: +- person +- bicycle +- car +- motorcycle +- airplane +- bus +- train +- truck +- boat +- traffic light +- fire hydrant +- stop sign +- parking meter +- bench +- bird +- cat +- dog +- horse +- sheep +- cow +- elephant +- bear +- zebra +- giraffe +- backpack +- umbrella +- handbag +- tie +- suitcase +- frisbee +- skis +- snowboard +- sports ball +- kite +- baseball bat +- baseball glove +- skateboard +- surfboard +- tennis racket +- bottle +- wine glass +- cup +- fork +- knife +- spoon +- bowl +- banana +- apple +- sandwich +- orange +- broccoli +- carrot +- hot dog +- pizza +- donut +- cake +- chair +- couch +- potted plant +- bed +- dining table +- toilet +- tv +- laptop +- mouse +- remote +- keyboard +- cell phone +- microwave +- oven +- toaster +- sink +- refrigerator +- book +- clock +- vase +- scissors +- teddy bear +- hair drier +- toothbrush diff --git a/models/smoking_detection_paddle/model.pdiparams b/models/smoking_detection_paddle/model.pdiparams new file mode 100644 index 0000000..3910efa --- /dev/null +++ b/models/smoking_detection_paddle/model.pdiparams @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3136b301bc46098aa7fbe3511a6919cb885ca1c8f8070ec73ad573c05eac4db9 +size 29968179 diff --git a/models/smoking_detection_paddle/model.pdiparams.info b/models/smoking_detection_paddle/model.pdiparams.info new file mode 100644 index 0000000..6075365 Binary files /dev/null and b/models/smoking_detection_paddle/model.pdiparams.info differ diff --git a/models/smoking_detection_paddle/model.pdmodel b/models/smoking_detection_paddle/model.pdmodel new file mode 100644 index 0000000..0e83160 Binary files /dev/null and b/models/smoking_detection_paddle/model.pdmodel differ