Skip to content

ProxyAyush/SortEase

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

24 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Python PyTorch OpenCV Visual Studio Code Windows 11 Freelancer


DOI

SortEase

WHERE BIOMEDICAL WASTE FINDS ITS PLACE​

This Model is based on Yolov8s and trained using a custom image dataset from King George's Medical University's Biomedical Waste Facility, Data of Biomedical Waste was collected with all necessary permissions from University Environment Department, and was taken in controlled manner with appropriate protection.

This is not production Quality AI Model, rather a proof of concept and It can detect Sharps which unfortunately come in Red Dustbins in Hospitals.
7 types of sharps it detects are "Blades-Sharp-", "Cannula_with_Cap -Sharp-", "Glass -Sharp-", "Instrument-Sharp-", "Needle -Sharp-", "Needle_with_Cap -Sharp-", "Syringes -Alert-".
One can use this for their testing on their PC, the following instructions guide you how to do that.

Download and Install Vscode(Visual Studio)

Download Link- https://code.visualstudio.com/download
Meanwhile Vscode is downloading, create an empty folder on your Desktop with any name you want.
Install Vscode after it downloads
After installing Vscode, Run it.
You'll see a 'Open Folder' option, select the empty folder you just created and Open it

Also install the python interpreter in vscode, on how to do that see this
https://code.visualstudio.com/docs/python/python-tutorial

Download Model and Inference file

Download the best.pt file
Download the liveinference.py
and place both these files in the empty folder you made earlier.

Installation of Libraries

We open the liveinference.py in Vscode(it shows in the folder panel), and on the upper right corner click the terminal button to open the terminal with code in Vscode
We install the Ultralytics Library in the terminal

py -m pip install ultralytics

then we will install the opencv library

py -m pip install opencv-python

Specifying the path in Inference file

In the Liveinference.py file opened in Vscode, check the model path in line number 10, change it to the path of best.pt file which you placed in your folder.

model = YOLO("YOUR PATH TO best.pt")

*there is already a path in liveinference.py, make sure to copy your path in the same way, or it'll give UNICODE escape error.

Running the Program

On the upper right side of Vscode there is a play button, click it and it will run the program with formation of a Webcam window.
Make sure you have Webcam enabled in your settings.


These are the training graphs of model:

graphs

Follow My Work on Linkedin

LinkedIn
https://www.linkedin.com/in/ayush-yadav-kgmu/?originalSubdomain=in

This project was made using the below resources:
Python PyTorch OpenCV Visual Studio Code Windows 11 Freelancer