Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

feat: Adds support of YOLOv4 detection model #78

Merged
merged 51 commits into from
Sep 2, 2020
Merged

feat: Adds support of YOLOv4 detection model #78

merged 51 commits into from
Sep 2, 2020

Conversation

frgfm
Copy link
Owner

@frgfm frgfm commented Aug 30, 2020

This PR extends detection model support to YOLOv4 by:

  • implementing YOLOv4 class
  • speeding up global average pooling for classification models
  • adding support of generalized IoU
  • updating Darknet models
  • updating unittests
  • updating documentation
  • updating readme

@frgfm
feat: Updated Darknet URLs
52c06e0
@frgfm
docs: Updated Readmes
9f71697
@frgfm
fix: Fixed backbone norm layer
688a1e0
@frgfm
feat: Added SPP & PAN layers
2a2e4c4
@frgfm
feat: Added YOLOv4 neck
0042e5b
@frgfm
style: Fixed lint
e6fb6f3
@frgfm
feat: Added Self Attention Module
a0109cf
@frgfm
docs: Fixed docstring
5dc89b8
@frgfm
style: Added extra_repr to GlobalAvgPool2d
38b2dc4
@frgfm
feat: Added implementation of YOLOv4
cc40f72
@frgfm
refactor: Updated addcdiv usage to reflect torch changes
cb07cb6
@frgfm
refactor: Switched anchors to buffer
37bfcac
@frgfm
style: Fixed lint
d79d76e
@frgfm
feat: Added compatibility for list of Tensor as input
3eb1bf9
@frgfm
fix: Passed down gt_boxes and gt_labels
b91aca8
@frgfm
style: Fixed lint
f3c299b
@frgfm
refactor: Reshaped model outputs
31373e8
@frgfm
feat: Added implementation of Generalized IoU
1ba700a
@frgfm
test: Updated unittests
7b9b359
@frgfm
docs: Updated documentation
fd74657
@frgfm
fix: Fixed YOLOv4 loss
2506b15
@frgfm
test: Updated unittests for models
da7d936
@frgfm
docs: Updated documentation
563937b
@frgfm
test: Added check to make sure loss can be backpropagated
f4abecd
@frgfm
fix: Fixed YOLO loss
a9e43a9
@frgfm
docs: Updated docstring
8f08d46
@frgfm
feat: Updated default backbone parameters
345b6d4
@frgfm
refactor: Refactored loss computation
c23d00c
@frgfm
refactor: Updated global avg poolings in models
47f9a04
@frgfm
style: Fixed lint
3c91feb
@frgfm frgfm added topic: documentation Improvements or additions to documentation type: enhancement New feature or request module: models ext: tests Related to test module: ops Related to ops labels Aug 30, 2020
@frgfm frgfm added this to the 0.1.3 milestone Aug 30, 2020
@frgfm frgfm self-assigned this Aug 30, 2020
@frgfm frgfm merged commit 5e54efa into master Sep 2, 2020
@frgfm frgfm deleted the yolov4 branch September 2, 2020 23:18
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees

@frgfm frgfm

Labels
ext: tests Related to test module: models module: ops Related to ops topic: documentation Improvements or additions to documentation type: enhancement New feature or request
Projects
None yet
Milestone
0.1.3
Development

Successfully merging this pull request may close these issues.

None yet
1 participant
@frgfm