search

UppuluriKalyani / ML-Nexus

ML Nexus is an open-source collection of machine learning projects, covering topics like neural networks, computer vision, and NLP. Whether you're a beginner or expert, contribute, collaborate, and grow together in the world of AI. Join us to shape the future of machine learning!
https://ml-nexus.vercel.app/
MIT License
62 stars 116 forks source link

Feature request Accident Classifier #41

Closed Pseudophoenix closed 1 month ago

Pseudophoenix commented 1 month ago

Is your feature request related to a problem? Please describe. Typically related to computer vision and transfer learning, I will be fine tuning the Yolov7 to classify images into severe accident and moderate accident

Describe the solution you'd like I will be using the yolov7 weights and then after freezing the base layer and modifying the upper layers to serve my purpose of accident severity classification.

Describe alternatives you've considered I also thought of instead going with ResNet50 fine tuning but Yolov7 seems a better option for now. I will try to go with ResNet as well through PyTorch.

Approach to be followed (optional) A clear and concise description of the approach to be followed.

Additional context I will be using this annotated Roboflow image dataset. https://universe.roboflow.com/accident-detection-model/accident-detection-model

Hope this will be assigned to me in the fine tuning section

github-actions[bot] commented 1 month ago

Thank you for creating this issue! 🎉 We'll look into it as soon as possible. In the meantime, please make sure to provide all the necessary details and context. Your contributions are highly appreciated! 😊

UppuluriKalyani commented 1 month ago

@Pseudophoenix go for it!

github-actions[bot] commented 1 month ago

Hello @Pseudophoenix! Your issue #41 has been closed. Thank you for your contribution!