adithya-s-k / World-of-AI

WORLD OF AI : An open-source repository for AI-based projects 🚀, from beginner to expert level, helping contributors start their journey in Artificial Intelligence and Deep Learning. Our projects provide hands-on experience to real-world problems👨‍💻. Join our community and contribute to the development of AI-based solutions 👥.
https://www.cognitivelab.tech/
MIT License
91 stars 86 forks source link

YOLOv8 Model Training on Custom Dataset Notebook #50

Open anupammaurya6767 opened 1 year ago

anupammaurya6767 commented 1 year ago

Project Description: We are seeking assistance in creating a Jupyter notebook for training the YOLOv8 object detection model on a custom dataset. YOLOv8 is a state-of-the-art deep learning model known for its accuracy and real-time object detection capabilities. By training the model on our custom dataset, we aim to achieve accurate and efficient detection of specific objects relevant to our application.

Project Goals:

  1. Develop a Jupyter notebook that facilitates training the YOLOv8 model on a custom dataset.
  2. Enable the customization of hyperparameters, such as learning rate, batch size, and training epochs, to optimize model performance.
  3. Provide support for data preprocessing, including annotation creation and bounding box adjustments.
  4. Implement transfer learning from pre-trained weights to expedite training and improve convergence.
  5. Incorporate data augmentation techniques to enhance the model's ability to generalize and handle variations in object appearance.
  6. Integrate model evaluation metrics to assess detection performance, including precision, recall, and mean average precision (mAP).
  7. Include instructions and documentation within the notebook to guide users through the entire training process, from data preparation to model evaluation.
anupammaurya6767 commented 1 year ago

@adithya-s-k Pls assign this issue to me I am a GSSOC 23 contributor I want to work on this issue

miraj0507 commented 1 year ago

Please do follow the project readme template kindly fix the template so that I can proceed further