/Custom_Object_Segmentaion_YOLOv8

Segment weapons and non-weapons in an image using YOLO v8

Primary LanguageJupyter Notebook

Custom_Object_Segmentaion_YOLOv8

Object segmentation using YOLO v8 by Ultralytics

setup environment

conda create -n ultralytics_yolo python=3.8
conda activate ultralytics_yolo
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116
pip install ultralytics
conda install jupyter
pip install cchardet
pip install chardet
pip install label-studio

(the alternate of the abobe pip installation is to clone their github repo:- https://github.com/ultralytics/ultralytics)