Add smoking detection Paddle model with LFS for large parameters
This commit is contained in:
1
.gitattributes
vendored
1
.gitattributes
vendored
@@ -1 +1,2 @@
|
|||||||
*.pt filter=lfs diff=lfs merge=lfs -text
|
*.pt filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.pdiparams filter=lfs diff=lfs merge=lfs -text
|
||||||
|
|||||||
1
.gitignore
vendored
1
.gitignore
vendored
@@ -73,5 +73,4 @@ behavior_detection/
|
|||||||
fire_detection/
|
fire_detection/
|
||||||
safety/
|
safety/
|
||||||
yolov/
|
yolov/
|
||||||
models/
|
|
||||||
__pycache__/
|
__pycache__/
|
||||||
|
|||||||
26
models/smoking_detection_paddle/infer_cfg.yml
Normal file
26
models/smoking_detection_paddle/infer_cfg.yml
Normal file
@@ -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
|
||||||
105
models/smoking_detection_paddle/infer_cfg_backup.yml
Normal file
105
models/smoking_detection_paddle/infer_cfg_backup.yml
Normal file
@@ -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
|
||||||
BIN
models/smoking_detection_paddle/model.pdiparams
LFS
Normal file
BIN
models/smoking_detection_paddle/model.pdiparams
LFS
Normal file
Binary file not shown.
BIN
models/smoking_detection_paddle/model.pdiparams.info
Normal file
BIN
models/smoking_detection_paddle/model.pdiparams.info
Normal file
Binary file not shown.
BIN
models/smoking_detection_paddle/model.pdmodel
Normal file
BIN
models/smoking_detection_paddle/model.pdmodel
Normal file
Binary file not shown.
Reference in New Issue
Block a user