Skip to content

GitLab

Unverified Commit d1f01835 authored by srihari-humbarwadi's avatar srihari-humbarwadi
Browse files

fixed bug in dataloader horizontal flip

parent ce62ac15
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/vision/beta/projects/panoptic_maskrcnn/dataloaders/panoptic_maskrcnn_input.py
View file @ d1f01835
...@@ -228,8 +228,8 @@ class Parser(maskrcnn_input.Parser): ...@@ -228,8 +228,8 @@ class Parser(maskrcnn_input.Parser):
image = image_mask[:, :, :-1] image = image_mask[:, :, :-1]
data['image'] = image data['image'] = image
data['boxes'] = boxes data['groundtruth_boxes'] = boxes
data['masks'] = masks data['groundtruth_instance_masks'] = masks
image, labels = super(Parser, self)._parse_train_data(data) image, labels = super(Parser, self)._parse_train_data(data)
......
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