Skip to content

Commit 622c16c

Browse files
authored
Merge branch 'main' into main
2 parents 05eb772 + 579f5f5 commit 622c16c

24 files changed

+476
-334
lines changed

.circleci/config.yml

Lines changed: 63 additions & 63 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

.circleci/regenerate.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ def build_workflows(prefix="", filter_branch=None, upload=False, indentation=6,
3232
for os_type in ["linux", "macos", "win"]:
3333
python_versions = PYTHON_VERSIONS
3434
cu_versions_dict = {
35-
"linux": ["cpu", "cu102", "cu111", "cu113", "cu115", "rocm4.2", "rocm4.3.1"],
35+
"linux": ["cpu", "cu102", "cu111", "cu113", "cu115", "rocm4.3.1", "rocm4.5.2"],
3636
"win": ["cpu", "cu111", "cu113", "cu115"],
3737
"macos": ["cpu"],
3838
}

.github/process_commit.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@
4343
"module: video",
4444
"Perf",
4545
"Revert(ed)",
46+
"topic: build",
4647
}
4748

4849

references/detection/transforms.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -189,7 +189,7 @@ def forward(
189189
elif image.ndimension() == 2:
190190
image = image.unsqueeze(0)
191191

192-
if torch.rand(1) < self.p:
192+
if torch.rand(1) >= self.p:
193193
return image, target
194194

195195
orig_w, orig_h = F.get_image_size(image)
@@ -211,6 +211,7 @@ def forward(
211211

212212
image = F.pad(image, [left, top, right, bottom], fill=fill)
213213
if isinstance(image, torch.Tensor):
214+
# PyTorch's pad supports only integers on fill. So we need to overwrite the colour
214215
v = torch.tensor(self.fill, device=image.device, dtype=image.dtype).view(-1, 1, 1)
215216
image[..., :top, :] = image[..., :, :left] = image[..., (top + orig_h) :, :] = image[
216217
..., :, (left + orig_w) :

0 commit comments

Comments
 (0)