本文整理匯總了Python中torch.utils.data.permute方法的典型用法代碼示例。如果您正苦於以下問題：Python data.permute方法的具體用法？Python data.permute怎麽用？Python data.permute使用的例子？那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類torch.utils.data的用法示例。
在下文中一共展示了data.permute方法的2個代碼示例，這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚，您的評價將有助於係統推薦出更棒的Python代碼示例。
示例1: __getitem__
# 需要導入模塊: from torch.utils import data [as 別名]
# 或者: from torch.utils.data import permute [as 別名]
def __getitem__(self, index):
# get the anchor index for current sample index
# here we set the anchor index to the last one
# sample in this group
minibatch_db = [self._roidb[index]] # [self._roidb[index_ratio]]
blobs = get_minibatch(minibatch_db, self._num_classes)
np.random.shuffle(blobs['rois'])
rois = torch.from_numpy(blobs['rois'][:self.max_rois_size])
data = torch.from_numpy(blobs['data'])
labels = torch.from_numpy(blobs['labels'])
data_height, data_width = data.size(1), data.size(2)
data = data.permute(0, 3, 1, 2).contiguous().view(3, data_height, data_width)
info = torch.Tensor([rois.size(0), data_height, data_width])
return data, rois, labels, info 示例2: __getitem__
# 需要導入模塊: from torch.utils import data [as 別名]
# 或者: from torch.utils.data import permute [as 別名]
def __getitem__(self, index):
indexes = [index]
blobs = self.get_one_sample(indexes)
data = torch.from_numpy(blobs['data'])
im_info = torch.from_numpy(blobs['im_info'])
mem_size = torch.from_numpy(blobs['memory_size'])
# we need to random shuffle the bounding box.
data_height, data_width = data.size(1), data.size(2)
if self.phase == 'train':
# if the number of region is greater than 100 then random pick 100 regions
# this opt can make the used memory of GPUs more stable.
# only for train and val phase
np.random.shuffle(blobs['gt_boxes'])
if blobs['gt_boxes'].shape[0] > 100:
print('sampling regions from %d to %d' % (blobs['gt_boxes'].shape[0], 100))
blobs['gt_boxes'] = blobs['gt_boxes'][:100]
elif self.phase == 'eval':
# np.random.shuffle(blobs['gt_boxes'])
if blobs['gt_boxes'].shape[0] > 100:
print('sampling regions from %d to %d' % (blobs['gt_boxes'].shape[0], 100))
blobs['gt_boxes'] = blobs['gt_boxes'][:100]
else:
pass
# if self.args.with_global:
# Arr_ = self._get_adjmat_Arr(blobs['gt_boxes']) # 5*r*r
# Arr = torch.from_numpy(Arr_)
# else:
# Arr = 0.
gt_boxes = torch.from_numpy(blobs['gt_boxes'])
# permute trim_data to adapt to downstream processing
data = data.permute(0, 3, 1, 2).contiguous().view(3, data_height, data_width)
im_info = im_info.view(3)
return data, im_info, gt_boxes, mem_size, blobs['data'], blobs['gt_boxes']