khushalps/api_internal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

(20241010) Almost ready for deployment.

Browse Source
This commit is contained in:
khushal
2024-10-10 11:44:45 +05:30
parent 103e42baad 0450b2a445
commit e858118467
13 changed files with 87 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils_v2/ai/object_detection/yolo.py
+2 -2
View File
@@ -10,7 +10,7 @@
OBJECTIVE:
To provide a class to detect objects in images using YOLO models.
To provide a class to detect objects in images using YOLO behaviour_models.
REFERENCES:
@@ -166,7 +166,7 @@ class YoloDetect:
if __name__ == "__main__":
# model_file_path = os.path.join(constants.PROJECT_DIRECTORY, "ai", "yolo", "models", "yolov8x.pt")
# model_file_path = os.path.join(constants.PROJECT_DIRECTORY, "ai", "yolo", "behaviour_models", "yolov8x.pt")
model_file_path = r"/home/developer/PycharmProjects/utils/data/ai/models/hugging_face/object_detection/YOLOv10-Document-Layout-Analysis/yolov10x_best.pt"
# sample_image_path = r"/home/developer/Downloads/2_cats.jpg"
sample_image_path = r"/home/developer/Downloads/flattened_image.jpg"