-
-
Notifications
You must be signed in to change notification settings - Fork 16.5k
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.
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
New YOLOv5 Classification Models #8956
Merged
Merged
Changes from 1 commit
Commits
Show all changes
448 commits
Select commit
Hold shift + click to select a range
4602c01
Update
glenn-jocher 798cab5
Logger step fix: Increment step with epochs (#8654)
AyushExel 5d69e04
Update
glenn-jocher fceeb9a
Update
glenn-jocher bdb33b9
Update
glenn-jocher 5fdeba7
Update
glenn-jocher bc72ec0
Update
glenn-jocher 1b00861
Update
glenn-jocher 053ada1
Update
glenn-jocher 167406d
Update
glenn-jocher dbe92bb
Update
glenn-jocher 761c1d3
Update
glenn-jocher 07297c4
Update
glenn-jocher d80c2f5
Update
glenn-jocher 8610aa8
Update
glenn-jocher 0524acb
Update
glenn-jocher c31cfb4
Update
glenn-jocher 0610a5d
Update
glenn-jocher 1433d3b
Update
glenn-jocher b66d142
Update
glenn-jocher 3a7a404
Update
glenn-jocher d11fc06
Update
glenn-jocher 334f8bc
Update
glenn-jocher ebebb62
Update
glenn-jocher 8faadee
Update
glenn-jocher 0a5a0ab
Update
glenn-jocher 7cc91cc
Update
glenn-jocher a545207
Update
glenn-jocher 50aaf98
Update
glenn-jocher 1c2f432
Allow logging models from GenericLogger (#8676)
AyushExel 5760567
Update
glenn-jocher c2095e1
Merge remote-tracking branch 'origin/classifier' into classifier
glenn-jocher a551986
Update
glenn-jocher 6c9ff8c
Update
glenn-jocher 956879b
Update
glenn-jocher bfb5c1f
Update
glenn-jocher 5ed36ee
Update warning emojis
glenn-jocher 04d3e64
Update dataset download
glenn-jocher dcf5670
Update dataset download
glenn-jocher a86fdd0
Update
glenn-jocher 2a59208
Update
glenn-jocher b80dd7e
Update
glenn-jocher 1097af8
Update
glenn-jocher 4edd9b2
Update
glenn-jocher 9256d45
Update
glenn-jocher 1d72089
Update
glenn-jocher 73bb5ff
Update
glenn-jocher 0670b38
Update
glenn-jocher aa9184d
Update
glenn-jocher 3626508
Update
glenn-jocher aa32e13
Update
glenn-jocher 0e010ae
Update
glenn-jocher 1ecaf0b
Update
glenn-jocher 685503b
Update
glenn-jocher 101b21f
Update
glenn-jocher 97d1fa6
Update
glenn-jocher 19ffb4e
Update
glenn-jocher 95eca5d
Update
glenn-jocher fa44db7
Update
glenn-jocher b751ab2
Update
glenn-jocher ff85584
Update
glenn-jocher e1fb3e6
Pass imgsz to classify_transforms()
glenn-jocher 7a73663
Update
glenn-jocher e535213
Update
glenn-jocher f839244
Update
glenn-jocher 2c90896
Update
glenn-jocher a6d5521
Update
glenn-jocher 9dc5e6a
Update
glenn-jocher 1782d0c
Update
glenn-jocher 84d6bd7
Update
glenn-jocher a06efbd
Update
glenn-jocher 6432df0
Update
glenn-jocher d86bb55
Update
glenn-jocher 0d6490c
Update
glenn-jocher 12421e1
Update
glenn-jocher 5f3773b
Update
glenn-jocher 9db31ea
Update
glenn-jocher 0551a31
Update
glenn-jocher f12bc8a
Update
glenn-jocher 9d9e16f
Update
glenn-jocher 55f0fa8
Update
glenn-jocher fabac7c
Update
glenn-jocher e64ba0e
Update
glenn-jocher 1038734
Update
glenn-jocher 86dd4d9
Cos scheduler
glenn-jocher 2594685
Cos scheduler
glenn-jocher 0251d35
Remove unused args
glenn-jocher 1f7b95c
Update
glenn-jocher e88ba9d
Add seed
glenn-jocher f610079
Add seed
glenn-jocher 635ab3f
Update
glenn-jocher eea5731
Update
glenn-jocher ef749a1
Add run(), main()
glenn-jocher d2cde35
Merge master
glenn-jocher 8a0da6d
Merge master
glenn-jocher a92840a
Merge master
glenn-jocher 976d5c7
Merge master
glenn-jocher f43572e
Update
glenn-jocher cabbb30
Update
glenn-jocher 59b2353
Update
glenn-jocher c5b92f5
Update
glenn-jocher ebb9c99
Update
glenn-jocher 7c1225f
Update
glenn-jocher e24b47d
Update
glenn-jocher 8ac8830
Merge master
glenn-jocher ebb141a
fix
glenn-jocher e567764
Create YOLOv5 BaseModel class (#8829)
glenn-jocher 13c46bd
Add experiment
glenn-jocher 1e548f0
Merge master
glenn-jocher 3284fe7
Merge master
glenn-jocher 1e9994a
Merge master
glenn-jocher 903169a
Attach names
glenn-jocher 84cc9bf
Merge master
glenn-jocher 2291181
weight decay = 1e-4
glenn-jocher 524e989
weight decay = 5e-5
glenn-jocher 4d3fede
update smart_optimizer console printout
glenn-jocher 2fe1b97
fashion-mnist fix
glenn-jocher 6a01f58
Merge master
glenn-jocher 7114bb9
Merge master
glenn-jocher ef00f41
Merge master
glenn-jocher 01bc47f
Merge master
glenn-jocher 9c5a035
Update Table
glenn-jocher bd135ba
Update Table
glenn-jocher 012476b
Remove destroy process group
glenn-jocher 037fb37
Merge master
glenn-jocher a1e81df
add kwargs to forward()
glenn-jocher 9c17a70
fuse fix for resnet50
glenn-jocher 7132881
nc, names fix for resnet50
glenn-jocher a512ab0
nc, names fix for resnet50
glenn-jocher c2cab3a
ONNX CPU inference fix
glenn-jocher 6302731
revert
glenn-jocher 7a60338
cuda
glenn-jocher ea8d237
if augment or visualize
glenn-jocher 49720f7
if augment or visualize
glenn-jocher 3f5c692
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] f27df13
New smart_inference_mode()
glenn-jocher 0a07a83
New smart_inference_mode()
glenn-jocher 5fe56fc
Update README
glenn-jocher 7c93f2c
Refactor into /classify dir
glenn-jocher ea8729b
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] 0173fdf
comma fix
glenn-jocher 30e46ed
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] 1d961a4
reset defaults
glenn-jocher 9f666e3
Merge remote-tracking branch 'origin/classifier' into classifier
glenn-jocher 52b73c7
reset defaults
glenn-jocher d97848e
fix gpu predict
glenn-jocher 17564e1
warmup
glenn-jocher f1f71c9
ema half fix
glenn-jocher 26d3a1e
spacing
glenn-jocher 0e9ba3c
remove data
glenn-jocher 4a982ae
remove cache
glenn-jocher dada286
remove denormalize
glenn-jocher 03e7a97
save run settings
glenn-jocher ef20d65
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] bc969a7
verbose false on initial plots
glenn-jocher 4d03b79
Merge remote-tracking branch 'origin/classifier' into classifier
glenn-jocher fbff212
new save_yaml() function
glenn-jocher 59e1300
Update ci-testing.yml
glenn-jocher 61f635f
Path(data) CI fix
glenn-jocher e8ed7cd
Separate classification CI
glenn-jocher 379a936
fix val
glenn-jocher f81686d
fix val
glenn-jocher 3dd6ef3
fix val
glenn-jocher e378210
smartCrossEntropyLoss
glenn-jocher 56c5b8e
skip validation on hub load
glenn-jocher 0e5720b
autodownload with working dir root
glenn-jocher 22b9d4b
str(data)
glenn-jocher 986390d
Dataset usage example
glenn-jocher b456f51
im_show normalize
glenn-jocher b7f3016
im_show normalize
glenn-jocher e663ff4
add imagenet simple names to multibackend
glenn-jocher cb13e7f
Add validation speeds
glenn-jocher 92fd072
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] d864cda
24-space names
glenn-jocher eafe847
Merge remote-tracking branch 'origin/classifier' into classifier
glenn-jocher 4020446
Update bash scripts
glenn-jocher 044022b
Update permissions
glenn-jocher 3d2d112
Add bash script arguments
glenn-jocher b41070b
Merge branch 'master' into classifier
glenn-jocher 13ec336
remove verbose
glenn-jocher 95090dc
TRT data fix
glenn-jocher e749931
names generator fix
glenn-jocher 3e6b07a
optimize if names
glenn-jocher 5963a25
update usage
glenn-jocher d35d9f0
Add local loading
glenn-jocher cb8ddc0
Verbose=False
glenn-jocher 5a5bd4e
update names printing
glenn-jocher b3bd5a1
Add Usage examples
glenn-jocher 4086b0f
Add Usage examples
glenn-jocher 6d2ce08
Add Usage examples
glenn-jocher d212511
Add Usage examples
glenn-jocher 3951615
Merge master
glenn-jocher 1abaf42
named_children
glenn-jocher db2f7b3
reshape_classifier_outputs
glenn-jocher 5b2070b
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] 1338c6d
update
glenn-jocher ab97b5a
Merge remote-tracking branch 'origin/classifier' into classifier
glenn-jocher b088d52
update
glenn-jocher 6730731
fix CI
glenn-jocher fc3b0bb
fix incorrect class substitution
glenn-jocher 0ff0ec2
fix incorrect class substitution
glenn-jocher 72e5607
remove denormalize
glenn-jocher bf1dae5
ravel fix
glenn-jocher 5f64ebd
cleanup
glenn-jocher bdee2e4
update opt file printing
glenn-jocher fd388d6
update opt file printing
glenn-jocher 1a307b8
update defaults
glenn-jocher 4ab5ee6
Merge branch 'master' into classifier
glenn-jocher 6a347f3
add opt to checkpoint
glenn-jocher 48da61b
Merge remote-tracking branch 'origin/classifier' into classifier
glenn-jocher 26dfbe3
Add warning
glenn-jocher 8ee42ff
Merge branch 'master' into classifier
glenn-jocher b6d3018
Add comment
glenn-jocher 0c97910
Merge remote-tracking branch 'origin/classifier' into classifier
glenn-jocher cc60aef
Merge branch 'master' into classifier
glenn-jocher 04e84d6
plot half bug fix
glenn-jocher 9d9a2f1
Use NotImplementedError
glenn-jocher 44584ce
fix export shape report
glenn-jocher b26129b
Fix TRT load
glenn-jocher 7449066
cleanup CI
glenn-jocher c0f0b7b
profile comment
glenn-jocher 01806e3
CI fix
glenn-jocher 1d867fd
Add cls models
glenn-jocher 4b75256
Merge branch 'master' into classifier
glenn-jocher ebb99af
avoid inplace error
glenn-jocher 3652de4
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] 105c73a
Merge branch 'master' into classifier
glenn-jocher d1c84ae
Fix usage examples
glenn-jocher c99c051
Merge remote-tracking branch 'origin/classifier' into classifier
glenn-jocher 706ff96
Update README
glenn-jocher 6f4a1bb
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] b5f8748
Update README
glenn-jocher ca01d76
Merge remote-tracking branch 'origin/classifier' into classifier
glenn-jocher 0a8f489
Update README
glenn-jocher 33ae017
Update README
glenn-jocher 7c494aa
Update README
glenn-jocher dcdc38b
Update README
glenn-jocher 85289d4
Update README
glenn-jocher ef63af1
Update README
glenn-jocher 105184c
Update README
glenn-jocher 47bbaee
Update README
glenn-jocher 4719685
Update README
glenn-jocher 55d85a6
Update README
glenn-jocher e8ce124
Update README
glenn-jocher 78e1bf9
Update README
glenn-jocher d0a3e70
Update README
glenn-jocher 4b398df
Update README
glenn-jocher e90cc95
Merge branch 'master' into classifier
glenn-jocher 9a43979
Update README
glenn-jocher 4b231d4
Merge remote-tracking branch 'origin/classifier' into classifier
glenn-jocher File filter
Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Update README
- Loading branch information
commit 5fe56fcc9483d1403ce6868a5530ebb1403a9855
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@glenn-jocher I think you're already planning to make this section collapsible right?
How about this structure instead:
[code] Example command for classification training and inference
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Yes, the current README content needs updating. I'll work on that a little later. I've refactored classification into /classify now, will add classification CI today.