Object detection Model with inclined angle calculation feature.
It incorporates yolov8, opencv2, with custom built yolov8 training dataset.
- Create a new venv
python venv -m Your_Venv_Name /path/to/your/venv/location
- Activate venv
/path/to/your/venv/Scripts/Activate.ps1
- Clone this git repo
git clone https://github.com/imranadas/Object_Detection_AngleCalculation_Yolo_CV.git
- Install libraries from requirements.txt in active venv
pip install -r /path/to/requirements.txt
- Training Examples: edit train.py and use yolov8 pretrained model or train from strach pre-trained models are preferables for weight transfer and efficient model training.
/path/to/venv/Scripts/python.exe /path/to/repo/train.py
- Detection of Books in images: detection fucntions are available in detection.py, use them freely in conjugtion with draw bounding_boxes functions.
- Detection and Angle Calculation: Run angle_calc.py in active venv and pass PATH in the terminal.
/path/to/venv/Scripts/python.exe /path/to/repo/angle_calc.py
Proceed detection and calcualtion by pressing enter on the image preview.
Program will store results locally in CWD/results folder
Re-Run of program will delete pre-existing result folder in CWD.
certifi==2022.12.7
charset-normalizer==2.1.1
colorama==0.4.6
contourpy==1.1.1
cycler==0.12.1
filelock==3.9.0
fonttools==4.43.1
fsspec==2023.4.0
idna==3.4
Jinja2==3.1.2
kiwisolver==1.4.5
MarkupSafe==2.1.2
matplotlib==3.8.0
mpmath==1.3.0
networkx==3.0
numpy==1.24.1
opencv-python==4.8.1.78
packaging==23.2
pandas==2.1.1
Pillow==9.3.0
psutil==5.9.6
py-cpuinfo==9.0.0
pyparsing==3.1.1
python-dateutil==2.8.2
pytz==2023.3.post1
PyYAML==6.0.1
requests==2.28.1
scipy==1.11.3
seaborn==0.13.0
six==1.16.0
sympy==1.12
thop==0.1.1.post2209072238
torch==2.1.0+cu118
torchaudio==2.1.0+cu118
torchvision==0.16.0+cu118
tqdm==4.66.1
typing_extensions==4.4.0
tzdata==2023.3
ultralytics==8.0.199
urllib3==1.26.13