Giter VIP home page Giter VIP logo

yolov8-3d's Introduction

YOLOv8-3D (tracker)

Description:

YOLOv8-3D is a lightweight and user-friendly library designed for efficient 2D and 3D bounding box object detection in Advanced Driver Assistance Systems (ADAS). With its intuitive API and comprehensive features, EasyADAS makes it straightforward to integrate object detection capabilities into your ADAS projects.

YOLOv8-3D

This API supports for easy understanding and integrate 3D perception, systems can make more informed decisions and operate effectively in complex, real-world environments.

Key Features:

  • Simplified API: YOLOv8-3D provides a straightforward API that allows you to quickly implement object detection in your ADAS system.
  • 2D and 3D Bounding Boxes: Detect both 2D and 3D bounding boxes for accurate spatial understanding of objects.
  • Efficient Processing: Leverage optimized algorithms for fast and reliable object detection performance.
  • Flexible Integration: EasyADAS is designed to seamlessly integrate with existing ADAS systems and frameworks.
  • Comprehensive Documentation: Extensive documentation and examples ensure that you can get started quickly and easily.
  • Scalable: Scale your detection capabilities by leveraging EasyADAS in a variety of scenarios and environments.
  • supports all best and and state of the art CNN models and easy training setup.
  • added augmentations for 3d .
  • api supports Resnets, VGG nets, mobilenets, Efficientnets , MOT-Tracker

special features

augmentations for better training, automated backup training and results plot

Get started

Prerequisites

Usage

Installation

  1. Clone this repository.
  2. Install the required dependencies

Run

for training

conda create -n test1 python=3.10 -y
conda activate test1
pip install tensorflow

For more detailed tensorflow gpu installation instructions and options, refer to this documentation.

####### select model on train.py  ########
# select_model = 'resnet50'
# select_model ='resnet101'
# select_model = 'resnet152'
# select_model = 'vgg11'
# select_model = 'vgg16'
# select_model = 'vgg19'
# select_model = 'efficientnetb0'
# select_model = 'efficientnetb5'
select_model = 'mobilenetv2'
###[INFO] set num of iterations to run (train.py) on (run_train.sh) file /// this automatically saves training info for every 20 epochs.
bash run_train.sh

recommended new environment to infer models only on cpu

for testing

conda create -n test2 python=3.10 -y
conda activate test2
pip install tensorflow ultralytics
python demo.py

realtime BEV plot

set
## BEV_plot = True
## TracK = True

Contributing

Contributions are welcome! If you find any issues or have suggestions for improvements, please open an issue or submit a pull request.

Acknowledgements

Expand

yolov8-3d's People

Contributors

bharath5673 avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar  avatar

yolov8-3d's Issues

Pretrained weights and more training details please?

Hi this repo looks interesting. Could you please provide pretrained weights and some sample data to run inference please. Also please add some more details of training, like what data to be downloaded from kitti, I see multiple links there. What is the folder structure needed?

Reference Paper

Hello @bharath5673
Thanks for building the repo.
I am trying to reproduce and make a version of this in Pytorch.
Would like to know if you used any paper as reference for this repo's approach.
I am also having some doubts on some parts of the approach.
Can you explain some of your ideas to me?
Thank you very much.

Performance difference

i have run your repo but i got different results regarding the performance. Could you tell me how to get a better performance
mobilenetv2_results_plot

Running demo.py with other videos

When I try to change sample video and run demo.py 2D boxes work quite well in detecting cars but 3D doesn't work at all.
I've run this video on mobilenetv2.
image
These are the primary parameters of this new video:
image
I have also tested this on vgg19 with the same result.
How should I add new demo video to this model?
Is this problem related to camera parameters? If so, where can I change them?

Problems with executing demo.py

Where can I get weights with the suffix "_weights.h5"? I download this weight and run demo.py with "ValueError: No model config found in the file at <tensorflow.python.platform.gfile.GFile object at 0x00000163B1C45CD0>.” How to solve it? Thank you very much for your reply. Thank you.

Custom Dataset

Hi, I just curious, is it possible to apply your work using custom dataset? In this case, the custom dataset already has the same structure as the KITTI dataset

ValueError: could not convert string to float: 'DontCare'

from train import *
  File "/media/ekin/Media7/berat-projects/YOLOv8-3D/train.py", line 129, in <module>
    all_objs, dims_avg = parse_annotation(label_dir, image_dir)
  File "/media/ekin/Media7/berat-projects/YOLOv8-3D/train.py", line 103, in parse_annotation
    occluded  = np.abs(float(line[2]))
ValueError: could not convert string to float: 'DontCare'

How is the organization of the dataset? I got that error

Issues while running

DNN library is not found.
	 [[{{node model/Conv1/Conv2D}}]] [Op:__inference_predict_function_10189]


Num GPUs Available:  1
loading file ...mobilenetv2_weights.h5...!
Traceback (most recent call last):
  File "/home/<>/Desktop/POINTCLOUD/YOLOv8-3D/demo.py", line 116, in <module>
    prediction = bbox3d_model.predict(patch, verbose = 0)
  File "/home/<>/anaconda3/envs/test2/lib/python3.10/site-packages/keras/src/utils/traceback_utils.py", line 70, in error_handler
    raise e.with_traceback(filtered_tb) from None
  File "/home/<>/anaconda3/envs/test2/lib/python3.10/site-packages/tensorflow/python/eager/execute.py", line 60, in quick_execute
    tensors = pywrap_tfe.TFE_Py_Execute(ctx._handle, device_name, op_name,
tensorflow.python.framework.errors_impl.UnimplementedError: Graph execution error:

Detected at node model/Conv1/Conv2D defined at (most recent call last):

Question on Inferencing Capabilities

Hello, I am interested in this work. Is this capable of generating 3d bounding boxes for other vehicles aside from the Kitti Dataset’s “car” class?

Many thanks!

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.