Skip to content

Commit

Permalink
Merge pull request #3 from simran29aw/main
Browse files Browse the repository at this point in the history
pull request
  • Loading branch information
simran29aw authored Sep 22, 2021
2 parents 4c25b25 + 2b1a8e6 commit dad1b65
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -13,8 +13,8 @@ The three main parts of our architecture are:
- Model Backbone - CSPNet are used as a backbone to extract rich informative features from an input image
- Model Neck - Model Neck is mainly used to generate feature pyramids. PANet is used for as neck to get feature pyramids
- Model Head - The model Head is mainly used to perform the final detection part. It applied anchor boxes on features and generates final output vectors with class probabilities, objectness scores, and bounding boxes.
<p align="center">
<img src="![The-network-architecture-of-Yolov5-It-consists-of-three-parts-1-Backbone-CSPDarknet](https://user-images.githubusercontent.com/72155378/134270700-b9005db1-2061-4508-a877-ef1362c3e0d4.jpg)" height="400" alt="Model structure">

![The-network-architecture-of-Yolov5-It-consists-of-three-parts-1-Backbone-CSPDarknet](https://user-images.githubusercontent.com/72155378/134271959-55ad63a4-ef1a-40fc-9c04-9e2369e19aa3.jpg)

The three main tasks of our project are:
- Object Detection using the YOLOv5 architecture
Expand Down Expand Up @@ -44,7 +44,7 @@ To use the repo and run inferences, please follow the guidelines below:
- PyTorch
- Numpy
- Pandas
- Opencv 4.5.1
- Opencv

## Contributors
- [Simran Agarwal](https://github.com/simran29aw)
Expand Down

0 comments on commit dad1b65

Please sign in to comment.