-
Notifications
You must be signed in to change notification settings - Fork 7.1k
add FCOS #4961
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
Merged
Merged
add FCOS #4961
Changes from all commits
Commits
Show all changes
94 commits
Select commit
Hold shift + click to select a range
791d938
add fcos
xiaohu2015 c9a00c1
update fcos
xiaohu2015 6c31577
Merge branch 'pytorch:main' into main
xiaohu2015 e82f825
add giou_loss
xiaohu2015 391b8c9
add BoxLinearCoder for FCOS
xiaohu2015 b01f145
add full code for FCOS
xiaohu2015 a4eeb59
add giou loss
xiaohu2015 61e946d
add fcos
xiaohu2015 997f189
add __all__
xiaohu2015 19ede68
Merge branch 'main' into main
xiaohu2015 793b1db
Fixing lint
zhiqwang a683060
Merge pull request #1 from o295/main
xiaohu2015 3cf91e3
Fixing lint in giou_loss.py
zhiqwang 1f67950
Merge pull request #2 from o295/main
xiaohu2015 5d75049
Merge branch 'pytorch:main' into main
xiaohu2015 c7a7d52
Add typing annotation to fcos
zhiqwang 42b147d
Add trained checkpoints
zhiqwang ff2d78a
Use partial to replace lambda
zhiqwang d4c08d3
Minor fixes to docstrings
zhiqwang edb2a1a
Merge pull request #3 from o295/main
xiaohu2015 c464249
Apply ufmt format
zhiqwang 8955e77
Merge pull request #4 from o295/main
xiaohu2015 293e6b9
Fixing docstrings
zhiqwang 4cbce22
Fixing jit scripting
zhiqwang b444d21
Minor fixes to docstrings
zhiqwang ac8d062
Fixing jit scripting
zhiqwang 795a550
Merge pull request #5 from o295/main
xiaohu2015 7b1c73f
Ignore mypy in fcos
zhiqwang 35702fd
Merge pull request #6 from o295/main
xiaohu2015 06e7ea8
Fixing trained checkpoints
zhiqwang 635a189
Merge pull request #7 from o295/main
xiaohu2015 47feb9c
Merge branch 'main' into main
xiaohu2015 d348da3
Fixing unit-test of jit script
zhiqwang 1802ca2
Fixing docstrings
zhiqwang 63c97e8
Merge pull request #8 from o295/main
xiaohu2015 9286ccd
Merge branch 'pytorch:main' into main
xiaohu2015 9f5034d
Add test/expect/ModelTester.test_fcos_resnet50_fpn_expect.pkl
zhiqwang 332af3b
Merge pull request #9 from o295/main
xiaohu2015 cfe224c
Fixing test_detection_model_trainable_backbone_layers
zhiqwang 9fada56
Merge pull request #10 from o295/main
xiaohu2015 36cb20b
Merge branch 'pytorch:main' into main
xiaohu2015 a729e57
Update test_fcos_resnet50_fpn_expect.pkl
zhiqwang a3caf0c
Merge pull request #11 from o295/main
xiaohu2015 570b147
Merge branch 'pytorch:main' into main
xiaohu2015 64c8613
Merge branch 'pytorch:main' into main
xiaohu2015 719c776
Merge branch 'pytorch:main' into main
xiaohu2015 c2c7a7e
rename stride to box size
xiaohu2015 7a830e1
remove TODO and fix some typo
xiaohu2015 b75d07b
Merge branch 'pytorch:main' into main
xiaohu2015 90efd29
merge some code for better
xiaohu2015 0e6039b
impove the comments
xiaohu2015 a0907f9
Merge branch 'pytorch:main' into main
xiaohu2015 4f9f3ef
remove decode and encode of BoxLinearCoder
xiaohu2015 db2e89b
remove some unnecessary hints
xiaohu2015 f5643ac
Merge branch 'pytorch:main' into main
xiaohu2015 d03f03a
use the default value in detectron2.
xiaohu2015 56896e5
update doc
xiaohu2015 fab2748
Add unittest for BoxLinearCoder
zhiqwang 80c3e64
Add types in FCOS
zhiqwang cb1f8e2
Add docstring for BoxLinearCoder
zhiqwang 56b131f
Minor fix for the docstring
zhiqwang ba328c0
Merge pull request #12 from o295/main
xiaohu2015 11cc5d9
update doc
xiaohu2015 e497021
Merge branch 'pytorch:main' into main
xiaohu2015 78e27da
Update fcos_resnet50_fpn_coco pretained weights url
jdsgomes 32ec6ba
Update torchvision/models/detection/fcos.py
xiaohu2015 0e9bd78
Update torchvision/models/detection/fcos.py
xiaohu2015 4faf24a
Update torchvision/models/detection/fcos.py
xiaohu2015 868fe54
Update torchvision/models/detection/fcos.py
xiaohu2015 a09507c
Merge branch 'pytorch:main' into main
xiaohu2015 54b12c8
Add FCOS model documentation
jdsgomes 6862094
Merge branch 'main' of github.com:xiaohu2015/vision into main
jdsgomes 589a3b1
Fix typo in FCOS documentation
jdsgomes a4db0dd
Add fcos to the prototype builder
zhiqwang 50bf19b
Capitalize COCO_V1
zhiqwang 688f4d2
Fix params of fcos
zhiqwang d4e04cf
Merge pull request #13 from zhiqwang/fcos-prototype-api
xiaohu2015 c75067f
Merge branch 'pytorch:main' into main
xiaohu2015 f01dcf6
fix bug for partial
xiaohu2015 9085461
Fixing docs indentation
zhiqwang 9b25135
Merge pull request #14 from zhiqwang/fix-doc-indentation
xiaohu2015 ebbd772
Fixing docs format in giou_loss
zhiqwang d8ed195
Adopt Reference for GIoU Loss
zhiqwang b125f66
Merge pull request #15 from zhiqwang/fix-lint
xiaohu2015 b0b0892
Rename giou_loss to generalized_box_iou_loss
jdsgomes 8c24ac3
remove overwrite_eps
jdsgomes bd3e262
Update AP test values
jdsgomes e452cfe
Minor fixes for the docs
zhiqwang e0af329
Minor fixes for the docs
zhiqwang c17a6e8
Merge pull request #17 from zhiqwang/remove_overwrite_eps
xiaohu2015 10be35d
Update torchvision/models/detection/fcos.py
xiaohu2015 4922635
Update torchvision/prototype/models/detection/fcos.py
xiaohu2015 6bae6c1
Merge branch 'main' into main
jdsgomes 6840a7d
Merge branch 'main' into main
jdsgomes 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
There are no files selected for viewing
This file contains hidden or 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
This file contains hidden or 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
This file contains hidden or 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
Binary file not shown.
This file contains hidden or 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
This file contains hidden or 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
This file contains hidden or 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
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -4,3 +4,4 @@ | |
from .retinanet import * | ||
from .ssd import * | ||
from .ssdlite import * | ||
from .fcos import * |
This file contains hidden or 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
Oops, something went wrong.
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.
Uh oh!
There was an error while loading. Please reload this page.