/med-drink-or-dose

build during Botnoi internship

Primary LanguagePython


Logo

Med Object Detection: Drink or Dose

Build during Botnoi internship

Table of Contents
  1. About The Project
  2. Getting Started
  3. Usage
  4. Roadmap
  5. Contact

About The Project

Using AI to detect an object in the video and recognize the object whether ti is drink or dose.

Dataset

Detail

  • Data Format: Video.mp4
  • Tool for labeling: Roboflow
  • Number of Classes: 2, Drink and Dose

Data Split

  • Train: 648
  • Valid: 176
  • Test: 115

(back to top)

Built With

  • Google Colab
  • VS Code
  • Python
  • Streamlit
  • OpenCV
  • PyTorch
  • YOLO Badge

(back to top)

Getting Started

Installation

  1. Clone the repo
git clone https://github.com/pangineering/med-drink-or-dose.git
  1. Install libraries
pip3 install -r requiorements.txt

(back to top)

Usage

to use with video

python3 app.py --media "video" --video "sample_2.mp4"

to use real time

 python3 app.py --media "live"

(back to top)

Roadmap

  • Live detection on the webcam
  • Detection on video

(back to top)

Contact

Gmail Badge Twitter Badge Facebook Badge Instagram Badge

(back to top)


BuyMeACoffee

Ko-Fi PayPal