Samarth991 commited on
Commit
46ea120
·
1 Parent(s): fa17459

added object detection imporved

Browse files
Files changed (2) hide show
  1. .gitignore +0 -19
  2. tool_utils/yolo_world.py +4 -4
.gitignore CHANGED
@@ -100,19 +100,6 @@ ipython_config.py
100
  # install all needed dependencies.
101
  #Pipfile.lock
102
 
103
- # poetry
104
- # Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
105
- # This is especially recommended for binary packages to ensure reproducibility, and is more
106
- # commonly ignored for libraries.
107
- # https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
108
- #poetry.lock
109
-
110
- # pdm
111
- # Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
112
- #pdm.lock
113
- # pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
114
- # in version control.
115
- # https://pdm.fming.dev/latest/usage/project/#working-with-version-control
116
  .pdm.toml
117
  .pdm-python
118
  .pdm-build/
@@ -160,11 +147,5 @@ dmypy.json
160
  # Cython debug symbols
161
  cython_debug/
162
 
163
- # PyCharm
164
- # JetBrains specific template is maintained in a separate JetBrains.gitignore that can
165
- # be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
166
- # and can be added to the global gitignore or merged into this file. For a more nuclear
167
- # option (not recommended) you can uncomment the following to ignore the entire idea folder.
168
- #.idea/
169
  *.json
170
  image_store/
 
100
  # install all needed dependencies.
101
  #Pipfile.lock
102
 
 
 
 
 
 
 
 
 
 
 
 
 
 
103
  .pdm.toml
104
  .pdm-python
105
  .pdm-build/
 
147
  # Cython debug symbols
148
  cython_debug/
149
 
 
 
 
 
 
 
150
  *.json
151
  image_store/
tool_utils/yolo_world.py CHANGED
@@ -85,8 +85,8 @@ class YoloWorld:
85
  cv2.imwrite('final_mask.png', cv2.cvtColor(detected_image,cv2.COLOR_BGR2RGB))
86
  return "Predicted image : final_mask.jpg . Details :{}".format(processed_predictions[0])
87
 
88
- if __name__ == "__main__":
89
- yolo = YoloWorld()
90
- predicted_data = yolo.run_yolo_infer('../image_store/demo2.jpg',['person','hat','building'])
91
- print(predicted_data)
92
 
 
85
  cv2.imwrite('final_mask.png', cv2.cvtColor(detected_image,cv2.COLOR_BGR2RGB))
86
  return "Predicted image : final_mask.jpg . Details :{}".format(processed_predictions[0])
87
 
88
+ # if __name__ == "__main__":
89
+ # yolo = YoloWorld()
90
+ # predicted_data = yolo.run_yolo_infer('../image_store/demo2.jpg',['person','hat','building'])
91
+ # print(predicted_data)
92