Charles Hernandez
43c910f431
fixing image_encoder to work with cuda_graphs
...
Summary: the combination of tensors on multiple devices in get_rel_pos
was preventing cuda graphs from correctly optimizing things
Test Plan:
Reviewers:
Subscribers:
Tasks:
Tags:
ghstack-source-id: 2256f130bb8249403710e1048ef69385ff71aed2
Pull Request resolved: https://github.com/facebookresearch/segment-anything/pull/393
2023-05-30 20:37:47 +00:00
Eric Mintun
567662b0fd
Fix incorrect shape in ResizeLongestSide.apply_image_torch.
2023-04-15 10:40:48 -07:00
anh-vunguyen
c1910835a3
Corrected docstring typo of class PatchEmbed
2023-04-12 11:55:29 -07:00
Murtaza rizvi
150e5812c7
Fixed Typo
2023-04-12 11:50:28 -07:00
Eric Mintun
dec4c12940
Fix mypy thinking output of SamPredictor.predict is a tensor.
2023-04-12 09:50:56 -07:00
derekray311511
d4ecc68dea
Update return type hint of predict function
2023-04-12 09:50:19 -07:00
Eric Mintun
7fa17d78c4
Lint.
2023-04-10 12:02:02 -07:00
Louis Maddox
b028d54358
Fix another instance of the same bug
2023-04-10 11:58:44 -07:00
Louis Maddox
6325eb8512
Move batched NMS indices to correct device ( closes #17 )
2023-04-10 11:58:44 -07:00
Eric Mintun
06bd20da89
Fix lint.
2023-04-10 09:21:25 -07:00
ericmintun
ca981bf377
Merge pull request #75 from Xrenya/patch-1
...
Update docstring of image_encoder.py
2023-04-10 08:54:57 -07:00
ericmintun
322eebc06f
Merge pull request #57 from Elm-Forest/main
...
Fixed some typos in comments
2023-04-10 08:49:32 -07:00
ericmintun
f3d37bde02
Merge pull request #37 from eltociear/patch-1
...
fix typo in mask_decoder.py
2023-04-10 08:44:41 -07:00
Hanzi Mao
f58f1d1cd1
Merge pull request #58 from EndingCredits/main
...
Fix baking original input size values to ONNX graph
2023-04-09 14:40:51 -07:00
Elm Forest
ea0f68a6c0
Merge branch 'facebookresearch:main' into main
2023-04-09 19:16:56 +08:00
Hanzi Mao
8991c14de4
remove direct call to build_sam()
2023-04-07 15:18:56 -07:00
Rinat
a77cb1fda9
Update docstring of image_encoder.py
...
minor docstring fix
2023-04-07 16:49:32 +09:00
William Woof
e9f2d58094
Update onnx.py
...
Fix break in computation graph due to typecast to python `int` instead of torch type.
2023-04-06 18:13:24 +01:00
Elm Forest
20114f5138
Fixed some typos in comments
2023-04-07 00:50:18 +08:00
Ikko Eltociear Ashimine
b7f0bc74a7
fix typo in mask_decoder.py
...
tranformer -> transformer
2023-04-06 17:11:56 +09:00
Eric Mintun
571794162e
Initial commit
2023-04-04 22:25:49 -07:00
