Upload 6 files
Browse files- YoloV8_Sam.ipynb +0 -0
- detectObjects.py +18 -0
- packages.txt +1 -0
- requirements.txt +129 -0
- sam_vit_b_01ec64.pth +3 -0
- yolov8n.pt +3 -0
YoloV8_Sam.ipynb
ADDED
The diff for this file is too large to render.
See raw diff
|
|
detectObjects.py
ADDED
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import ultralytics
|
2 |
+
from ultralytics import YOLO
|
3 |
+
|
4 |
+
model = YOLO('yolov8n.pt')
|
5 |
+
|
6 |
+
def detected_objects(filename:str):
|
7 |
+
results = model.predict(source=filename, conf=0.25)
|
8 |
+
|
9 |
+
categories = results[0].names
|
10 |
+
|
11 |
+
dc = []
|
12 |
+
for i in range(len(results[0])):
|
13 |
+
cat = results[0].boxes[i].cls
|
14 |
+
dc.append(categories[int(cat)])
|
15 |
+
|
16 |
+
print(dc)
|
17 |
+
return results, dc
|
18 |
+
|
packages.txt
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
libgl1
|
requirements.txt
ADDED
@@ -0,0 +1,129 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# altair==5.4.0
|
2 |
+
# asttokens==2.4.1
|
3 |
+
# attrs==24.2.0
|
4 |
+
# blinker==1.8.2
|
5 |
+
# cachetools==5.5.0
|
6 |
+
# certifi==2024.7.4
|
7 |
+
# charset-normalizer==3.3.2
|
8 |
+
# click==8.1.7
|
9 |
+
# coloredlogs==15.0.1
|
10 |
+
# comm==0.2.1
|
11 |
+
# contourpy==1.2.1
|
12 |
+
# cycler==0.12.1
|
13 |
+
# debugpy==1.8.0
|
14 |
+
# decorator==5.1.1
|
15 |
+
# # exceptiongroup @ file:///home/conda/feedstock_root/build_artifacts/exceptiongroup_1720869315914/work
|
16 |
+
# executing==2.0.1
|
17 |
+
# filelock==3.15.4
|
18 |
+
# flatbuffers==24.3.25
|
19 |
+
# fonttools==4.53.1
|
20 |
+
# fsspec==2024.6.1
|
21 |
+
# gitdb==4.0.11
|
22 |
+
# GitPython==3.1.43
|
23 |
+
# humanfriendly==10.0
|
24 |
+
# idna==3.8
|
25 |
+
# imageio==2.35.1
|
26 |
+
# # importlib_metadata @ file:///home/conda/feedstock_root/build_artifacts/importlib-metadata_1724187233579/work
|
27 |
+
# intel-openmp==2021.4.0
|
28 |
+
# ipykernel==6.29.0
|
29 |
+
# ipython==8.20.0
|
30 |
+
# jedi==0.19.1
|
31 |
+
# Jinja2==3.1.4
|
32 |
+
# jsonschema==4.23.0
|
33 |
+
# jsonschema-specifications==2023.12.1
|
34 |
+
# jupyter_client==8.6.0
|
35 |
+
# jupyter_core==5.7.1
|
36 |
+
# kiwisolver==1.4.5
|
37 |
+
# lazy_loader==0.4
|
38 |
+
# llvmlite==0.43.0
|
39 |
+
# markdown-it-py==3.0.0
|
40 |
+
# MarkupSafe==2.1.5
|
41 |
+
# matplotlib==3.9.2
|
42 |
+
# matplotlib-inline==0.1.6
|
43 |
+
# mdurl==0.1.2
|
44 |
+
# mkl==2021.4.0
|
45 |
+
# mpmath==1.3.0
|
46 |
+
# narwhals==1.5.5
|
47 |
+
# nest-asyncio==1.5.9
|
48 |
+
# networkx==3.3
|
49 |
+
# numba==0.60.0
|
50 |
+
# numpy==1.26.4
|
51 |
+
# onnxruntime==1.19.0
|
52 |
+
# opencv-contrib-python-headless==4.10.0.84
|
53 |
+
# opencv-python==4.10.0.84
|
54 |
+
# opencv-python-headless==4.10.0.84
|
55 |
+
# packaging==23.2
|
56 |
+
# pandas==2.2.2
|
57 |
+
# parso==0.8.3
|
58 |
+
# # pickleshare @ file:///home/conda/feedstock_root/build_artifacts/pickleshare_1602536217715/work
|
59 |
+
# pillow==10.4.0
|
60 |
+
# # platformdirs @ file:///home/conda/feedstock_root/build_artifacts/platformdirs_1715777629804/work
|
61 |
+
# pooch==1.8.2
|
62 |
+
# prompt-toolkit==3.0.43
|
63 |
+
# protobuf==5.27.3
|
64 |
+
# psutil==5.9.7
|
65 |
+
# pure-eval==0.2.2
|
66 |
+
# py-cpuinfo==9.0.0
|
67 |
+
# pyarrow==17.0.0
|
68 |
+
# pydeck==0.9.1
|
69 |
+
# Pygments==2.17.2
|
70 |
+
# PyMatting==1.1.12
|
71 |
+
# pyparsing==3.1.3
|
72 |
+
# pyreadline3==3.4.1
|
73 |
+
# python-dateutil==2.8.2
|
74 |
+
# pytz==2024.1
|
75 |
+
# pywin32==306
|
76 |
+
# PyYAML==6.0.2
|
77 |
+
# pyzmq==25.1.2
|
78 |
+
# referencing==0.35.1
|
79 |
+
# rembg==2.0.58
|
80 |
+
# requests==2.32.3
|
81 |
+
# rich==13.7.1
|
82 |
+
# rpds-py==0.20.0
|
83 |
+
# scikit-image==0.24.0
|
84 |
+
# scipy==1.14.1
|
85 |
+
# seaborn==0.13.2
|
86 |
+
# segment_anything @ git+https://github.com/facebookresearch/segment-anything.git@6fdee8f2727f4506cfbbe553e23b895e27956588
|
87 |
+
# six==1.16.0
|
88 |
+
# smmap==5.0.1
|
89 |
+
# stack-data==0.6.3
|
90 |
+
# streamlit==1.37.1
|
91 |
+
# sympy==1.13.2
|
92 |
+
# tbb==2021.13.1
|
93 |
+
# tenacity==8.5.0
|
94 |
+
# tifffile==2024.8.10
|
95 |
+
# toml==0.10.2
|
96 |
+
# torch==2.3.0
|
97 |
+
# torchvision==0.18.0
|
98 |
+
# tornado==6.4
|
99 |
+
# tqdm==4.66.5
|
100 |
+
# traitlets==5.14.1
|
101 |
+
# # typing_extensions @ file:///home/conda/feedstock_root/build_artifacts/typing_extensions_1717802530399/work
|
102 |
+
# tzdata==2024.1
|
103 |
+
# ultralytics==8.2.81
|
104 |
+
# ultralytics-thop==2.0.5
|
105 |
+
# urllib3==2.2.2
|
106 |
+
# watchdog==4.0.2
|
107 |
+
# wcwidth==0.2.13
|
108 |
+
# wget==3.2
|
109 |
+
# # zipp @ file:///home/conda/feedstock_root/build_artifacts/zipp_1723591248676/work
|
110 |
+
|
111 |
+
|
112 |
+
|
113 |
+
# New Requirements.txt file for Deployment
|
114 |
+
|
115 |
+
matplotlib
|
116 |
+
|
117 |
+
# opencv-python-headless
|
118 |
+
# opencv-contrib-python-headless==4.10.0.84
|
119 |
+
opencv-contrib-python-headless
|
120 |
+
# opencv-python==4.10.0.84
|
121 |
+
# opencv-python-headless==4.10.0.84
|
122 |
+
|
123 |
+
segment_anything @ git+https://github.com/facebookresearch/segment-anything.git@6fdee8f2727f4506cfbbe553e23b895e27956588
|
124 |
+
|
125 |
+
rembg
|
126 |
+
|
127 |
+
streamlit
|
128 |
+
|
129 |
+
ultralytics==8.2.81
|
sam_vit_b_01ec64.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ec2df62732614e57411cdcf32a23ffdf28910380d03139ee0f4fcbe91eb8c912
|
3 |
+
size 375042383
|
yolov8n.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f59b3d833e2ff32e194b5bb8e08d211dc7c5bdf144b90d2c8412c47ccfc83b36
|
3 |
+
size 6549796
|