"docs/source/en/optimization/fp16.md" did not exist on "8aac1f99d7af5873db7d23c07fba370d0f5061a6"
Commit e8e9bdb6 authored by buoyancy99's avatar buoyancy99 Committed by Francisco Massa
Browse files

fix transform for rcnns so original images list is unchanged (#1084)

* fix transform for rcnns so original images are unchanged

* transform does not change input list anymore

transform does not change input list anymore. Improve code according to reviewer comment

* transform for maskrcnn no longer modify input list

transform for maskrcnn no longer modify input list. improve code according to comment

* transform for maskrcnn no longer modifies input list
parent 7d8cc199
......@@ -30,6 +30,7 @@ class GeneralizedRCNNTransform(nn.Module):
self.image_std = image_std
def forward(self, images, targets=None):
images = images[:]
for i in range(len(images)):
image = images[i]
target = targets[i] if targets is not None else targets
......@@ -41,6 +42,7 @@ class GeneralizedRCNNTransform(nn.Module):
images[i] = image
if targets is not None:
targets[i] = target
image_sizes = [img.shape[-2:] for img in images]
images = self.batch_images(images)
image_list = ImageList(images, image_sizes)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment