In this repository is hosted the code for object detection and segmentation with YoloV8.
- In the model.py file the yolov8n-seg model is retrained with custom data.
- The predict.py file is used to test the segmentation model with images.
- Real-time inference is performed in the RtPredict.py file.
- The ONNXModel.py and TRTmodel.py files optimize the model with ONNX and TensorRT respectively.
- The files weights150e.pt (Yolo), ModelONNX.onnx (ONNX), ModelTRT.engine (TensorRT) contain the weights of the retrained model with the customized data.