Skip to content

Add LW-DETR object detection models #334

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

Merged
cregouby merged 9 commits into from
Jul 4, 2026
Merged

Add LW-DETR object detection models #334

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
5e173a9
Add LW-DETR object detection models
srishtiii28 Jun 24, 2026
3d11f48
Updated NEWS.md
srishtiii28 Jun 24, 2026
592c49b
Add models-lw_detr.R to Collate
srishtiii28 Jun 24, 2026
6de639d
Cross-link model_lw_detr in detection model docs
srishtiii28 Jun 24, 2026
2a32933
Add pixel_mask test for model_lw_detr
srishtiii28 Jun 24, 2026
c333e16
Updated NEWS.md to fix misplacement of model
srishtiii28 Jun 26, 2026
9ad1337
usethis::use_air()
cregouby Jun 28, 2026
11cd2af
Address review feedback for lw_detr model
srishtiii28 Jun 29, 2026
3ee49cc
cleanup IDE scories
cregouby Jul 4, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions .Rbuildignore
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -15,3 +15,5 @@
^CRAN-RELEASE$
^CRAN-SUBMISSION$
^data-raw$
^[.]?air[.]toml$
^\.vscode$
8 changes: 3 additions & 5 deletions .gitignore
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,10 +1,8 @@
.Rproj.user
models/
env/
test.pth
testing.pth
x.pth
docs
docs/
.Rhistory
s.pth
inst/doc
tools/
.vscode/
1 change: 1 addition & 0 deletions DESCRIPTION
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -93,6 +93,7 @@ Collate:
'models-faster_rcnn.R'
'models-fcn.R'
'models-inception.R'
'models-lw_detr.R'
'models-mask_rcnn.R'
'models-maxvit.R'
'models-mobilenetv2.R'
Expand Down
4 changes: 4 additions & 0 deletions NAMESPACE
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -154,6 +154,10 @@ export(model_fasterrcnn_resnet50_fpn_v2)
export(model_fcn_resnet101)
export(model_fcn_resnet50)
export(model_inception_v3)
export(model_lw_detr_large)
export(model_lw_detr_medium)
export(model_lw_detr_small)
export(model_lw_detr_tiny)
export(model_maskrcnn_resnet50_fpn)
export(model_maskrcnn_resnet50_fpn_v2)
export(model_maxvit)
Expand Down
4 changes: 4 additions & 0 deletions NEWS.md
View file
Open in desktop
Comment thread
srishtiii28 marked this conversation as resolved.
Original file line number Diff line number Diff line change
@@ -1,5 +1,9 @@
# torchvision (development version)

## New models

* Added `model_lw_detr_tiny()`, `model_lw_detr_small()`, `model_lw_detr_medium()` and `model_lw_detr_large()` for real-time object detection (@srishtiii28, #328).

## Bug fixes and improvements

* `nms()` now uses `torchvisionlib::ops_nms()` when torchvisionlib is installed, speeding up inference for `model_fasterrcnn_*()` and `model_maskrcnn_*()` (#321, #322).
Expand Down
Loading
Loading