[go: nahoru, domu]

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.

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

Add automatic-mask-generation pipeline for Segment Anything Model (SAM) #22840

Merged
merged 68 commits into from
Apr 20, 2023
Merged
Changes from 1 commit
Commits
Show all changes
68 commits
Select commit Hold shift + click to select a range
eb4dc26
cleanup
ArthurZucker Apr 18, 2023
f3f345b
updates
ArthurZucker Apr 19, 2023
94fd59d
more refactoring
ArthurZucker Apr 19, 2023
6bbb106
make style
ArthurZucker Apr 19, 2023
52675a6
update inits
ArthurZucker Apr 19, 2023
3901d13
support other inputs in base
ArthurZucker Apr 19, 2023
df77716
update based on review
ArthurZucker Apr 19, 2023
9d9e70b
Update tests/pipelines/test_pipelines_automatic_mask_generation.py
ArthurZucker Apr 19, 2023
0a7f702
update
ArthurZucker Apr 19, 2023
d57ed60
fixup
ArthurZucker Apr 19, 2023
2d11d6f
TODO x and y to refactor, _h _w refactored here
ArthurZucker Apr 19, 2023
8950318
update docstring
ArthurZucker Apr 19, 2023
c68453a
more nits
ArthurZucker Apr 19, 2023
e8b45b3
style on these
ArthurZucker Apr 19, 2023
ef08e55
more doc fix
ArthurZucker Apr 19, 2023
d9027ed
rename variables
younesbelkada Apr 19, 2023
0e507b8
update
ArthurZucker Apr 19, 2023
3b6f918
Merge branch 'main' of https://github.com/huggingface/transformers in…
ArthurZucker Apr 19, 2023
7dbdbc6
updates
ArthurZucker Apr 19, 2023
23898c1
style
ArthurZucker Apr 19, 2023
cff3a38
update
ArthurZucker Apr 19, 2023
a2e1038
fix `_mask_to_rle_pytorch`
ArthurZucker Apr 20, 2023
a4aac94
styling
ArthurZucker Apr 20, 2023
05b04c8
fix ask to rle, wrong outputs
ArthurZucker Apr 20, 2023
e5a97df
add device arg
younesbelkada Apr 20, 2023
b1591e5
update
ArthurZucker Apr 20, 2023
deee7fe
Merge branch 'add-mg-pipeline' of https://github.com/ArthurZucker/tra…
ArthurZucker Apr 20, 2023
ad88b20
more updates, fix tets
ArthurZucker Apr 20, 2023
b8f5980
udpate
ArthurZucker Apr 20, 2023
796ec85
update docstrings
ArthurZucker Apr 20, 2023
a98eb36
styling
ArthurZucker Apr 20, 2023
093d37d
fixup
ArthurZucker Apr 20, 2023
f64b444
add notebook on the docs
younesbelkada Apr 20, 2023
4fd393d
update orginal sizes
ArthurZucker Apr 20, 2023
2dbe327
Merge branch 'add-mg-pipeline' of https://github.com/ArthurZucker/tra…
ArthurZucker Apr 20, 2023
4c0412d
fix docstring
younesbelkada Apr 20, 2023
395e08b
updat condition on point_per-batch
ArthurZucker Apr 20, 2023
54e797c
Merge branch 'add-mg-pipeline' of https://github.com/ArthurZucker/tra…
younesbelkada Apr 20, 2023
8f9f07c
updates tests
ArthurZucker Apr 20, 2023
83f6a84
fix CI test
younesbelkada Apr 20, 2023
c647a50
extend is required, append does not work!
ArthurZucker Apr 20, 2023
04a6cf2
Merge branch 'add-mg-pipeline' of https://github.com/ArthurZucker/tra…
ArthurZucker Apr 20, 2023
bca34eb
fixup
ArthurZucker Apr 20, 2023
e121428
fix CI tests
younesbelkada Apr 20, 2023
3664c01
whit pixels left
ArthurZucker Apr 20, 2023
983d0e1
address doc comments
younesbelkada Apr 20, 2023
7dcdd36
fix doc
younesbelkada Apr 20, 2023
75b9576
slow pipeline tests
ArthurZucker Apr 20, 2023
342109b
update auto init
ArthurZucker Apr 20, 2023
df58947
Merge branch 'add-mg-pipeline' of https://github.com/ArthurZucker/tra…
ArthurZucker Apr 20, 2023
b9e4272
add revision
younesbelkada Apr 20, 2023
749b94f
Merge branch 'add-mg-pipeline' of https://github.com/ArthurZucker/tra…
younesbelkada Apr 20, 2023
d5eb3f9
make fixup
younesbelkada Apr 20, 2023
e103ef2
update p!ipoeline tag when calling tests
ArthurZucker Apr 20, 2023
ed4e727
alphabeitcal order in inits
ArthurZucker Apr 20, 2023
3c4f4b8
fix copies
ArthurZucker Apr 20, 2023
eb9bd0c
last style nits
ArthurZucker Apr 20, 2023
d98f0e3
Merge branch 'add-mg-pipeline' of https://github.com/ArthurZucker/tra…
ArthurZucker Apr 20, 2023
7a1f99a
Apply suggestions from code review
younesbelkada Apr 20, 2023
94a7af3
reformat docstring
younesbelkada Apr 20, 2023
7d8b860
more reformat
younesbelkada Apr 20, 2023
a59133a
address most of the comments
younesbelkada Apr 20, 2023
e970f30
Update src/transformers/pipelines/mask_generation.py
younesbelkada Apr 20, 2023
b5d56da
final refactor
younesbelkada Apr 20, 2023
32d6e80
Update src/transformers/models/sam/image_processing_sam.py
younesbelkada Apr 20, 2023
4149f36
fixup and fix slow tests
younesbelkada Apr 20, 2023
9a0c19d
Merge remote-tracking branch 'upstream/main' into HEAD
younesbelkada Apr 20, 2023
3b4a594
revert
younesbelkada Apr 20, 2023
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
Prev Previous commit
Next Next commit
style on these
  • Loading branch information
ArthurZucker committed Apr 19, 2023
commit e8b45b3cb5cfd48628775974efb026f7db726add
3 changes: 2 additions & 1 deletion src/transformers/pipelines/mask_generation.py
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,8 @@ class MaskGenerationPipeline(ChunkPipeline):
feature_extractor ([`SequenceFeatureExtractor`]):
The feature extractor that will be used by the pipeline to encode waveform for the model.
younesbelkada marked this conversation as resolved.
Show resolved Hide resolved
points_per_batch (*optional*, int, default to 64):
Sets the number of points run simultaneously by the model. heightigher numbers may be faster but use more GPU memory.
Sets the number of points run simultaneously by the model. heightigher numbers may be faster but use more
younesbelkada marked this conversation as resolved.
Show resolved Hide resolved
GPU memory.
output_bboxes_mask (`bool`, *optional*, default to `False`):
Whether or not to output the bounding box predictions.
output_rle_masks (`bool`, *optional*, default to `False`):
Expand Down