├── README.md
├── object detect
└── .gitignore


/README.md:
--------------------------------------------------------------------------------
1 | # save-workers-cognizent


--------------------------------------------------------------------------------
/object detect:
--------------------------------------------------------------------------------
 1 | def objectdetect():
 2 |     cap = cv2.VideoCapture(0)
 3 |     cap.set(3, 6430)
 4 |     cap.set(4, 480)
 5 | 
 6 |     classNames = []
 7 |     classFile = 'coco.names'
 8 |     with open(classFile, 'rt') as f:
 9 |         classNames = f.read().rstrip('\n').split('\n')
10 | 
11 |     configPath = 'ssd_mobilenet_v3_large_coco_2020_01_14.pbtxt'
12 |     weightsPath = 'frozen_inference_graph.pb'
13 | 
14 |     net = cv2.dnn_DetectionModel(weightsPath, configPath)
15 |     net.setInputSize(320, 320)
16 |     net.setInputScale(1.0 / 127.5)
17 |     net.setInputMean((127.5, 127.5, 127.5))
18 |     net.setInputSwapRB(True)
19 | 
20 |     while True:
21 |         success, img = cap.read()
22 |         classIds, confs, bbox = net.detect(img, confThreshold=0.5)
23 |         print(classIds, bbox)
24 | 
25 |         if len(classIds) != 0:
26 |             for classId, confidence, box in zip(classIds.flatten(), confs.flatten(), bbox):
27 |                 cv2.rectangle(img, box, color=(0, 255, 0), thickness=2)
28 |                 cv2.putText(img, classNames[classId - 1].upper(), (box[0] + 10, box[1] + 30), cv2.FONT_HERSHEY_COMPLEX,
29 |                             1, (0, 255, 0), 2)
30 |                 cv2.putText(img, str(round(confidence * 100, 2)), (box[0] + 200, box[1] + 30), cv2.FONT_HERSHEY_COMPLEX,
31 |                             1, (0, 255, 0), 2)
32 | 
33 |         cv2.imshow("Output", img)
34 |         cv2.waitKey(1)
35 | 


--------------------------------------------------------------------------------
/.gitignore:
--------------------------------------------------------------------------------
  1 | # Byte-compiled / optimized / DLL files
  2 | __pycache__/
  3 | *.py[cod]
  4 | *$py.class
  5 | 
  6 | # C extensions
  7 | *.so
  8 | 
  9 | # Distribution / packaging
 10 | .Python
 11 | build/
 12 | develop-eggs/
 13 | dist/
 14 | downloads/
 15 | eggs/
 16 | .eggs/
 17 | lib/
 18 | lib64/
 19 | parts/
 20 | sdist/
 21 | var/
 22 | wheels/
 23 | pip-wheel-metadata/
 24 | share/python-wheels/
 25 | *.egg-info/
 26 | .installed.cfg
 27 | *.egg
 28 | MANIFEST
 29 | 
 30 | # PyInstaller
 31 | #  Usually these files are written by a python script from a template
 32 | #  before PyInstaller builds the exe, so as to inject date/other infos into it.
 33 | *.manifest
 34 | *.spec
 35 | 
 36 | # Installer logs
 37 | pip-log.txt
 38 | pip-delete-this-directory.txt
 39 | 
 40 | # Unit test / coverage reports
 41 | htmlcov/
 42 | .tox/
 43 | .nox/
 44 | .coverage
 45 | .coverage.*
 46 | .cache
 47 | nosetests.xml
 48 | coverage.xml
 49 | *.cover
 50 | *.py,cover
 51 | .hypothesis/
 52 | .pytest_cache/
 53 | 
 54 | # Translations
 55 | *.mo
 56 | *.pot
 57 | 
 58 | # Django stuff:
 59 | *.log
 60 | local_settings.py
 61 | db.sqlite3
 62 | db.sqlite3-journal
 63 | 
 64 | # Flask stuff:
 65 | instance/
 66 | .webassets-cache
 67 | 
 68 | # Scrapy stuff:
 69 | .scrapy
 70 | 
 71 | # Sphinx documentation
 72 | docs/_build/
 73 | 
 74 | # PyBuilder
 75 | target/
 76 | 
 77 | # Jupyter Notebook
 78 | .ipynb_checkpoints
 79 | 
 80 | # IPython
 81 | profile_default/
 82 | ipython_config.py
 83 | 
 84 | # pyenv
 85 | .python-version
 86 | 
 87 | # pipenv
 88 | #   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
 89 | #   However, in case of collaboration, if having platform-specific dependencies or dependencies
 90 | #   having no cross-platform support, pipenv may install dependencies that don't work, or not
 91 | #   install all needed dependencies.
 92 | #Pipfile.lock
 93 | 
 94 | # PEP 582; used by e.g. github.com/David-OConnor/pyflow
 95 | __pypackages__/
 96 | 
 97 | # Celery stuff
 98 | celerybeat-schedule
 99 | celerybeat.pid
100 | 
101 | # SageMath parsed files
102 | *.sage.py
103 | 
104 | # Environments
105 | .env
106 | .venv
107 | env/
108 | venv/
109 | ENV/
110 | env.bak/
111 | venv.bak/
112 | 
113 | # Spyder project settings
114 | .spyderproject
115 | .spyproject
116 | 
117 | # Rope project settings
118 | .ropeproject
119 | 
120 | # mkdocs documentation
121 | /site
122 | 
123 | # mypy
124 | .mypy_cache/
125 | .dmypy.json
126 | dmypy.json
127 | 
128 | # Pyre type checker
129 | .pyre/
130 | 


--------------------------------------------------------------------------------