本文整理汇总了Python中mmdet.datasets.to_tensor方法的典型用法代码示例。如果您正苦于以下问题：Python datasets.to_tensor方法的具体用法？Python datasets.to_tensor怎么用？Python datasets.to_tensor使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类mmdet.datasets的用法示例。
在下文中一共展示了datasets.to_tensor方法的2个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: _prepare_data
# 需要导入模块: from mmdet import datasets [as 别名]
# 或者: from mmdet.datasets import to_tensor [as 别名]
def _prepare_data(img, img_transform, cfg, device):
ori_shape = img.shape
img, img_shape, pad_shape, scale_factor = img_transform(
img,
scale=cfg.data.test.img_scale,
keep_ratio=cfg.data.test.get('resize_keep_ratio', True))
img = to_tensor(img).to(device).unsqueeze(0)
img_meta = [
dict(
ori_shape=ori_shape,
img_shape=img_shape,
pad_shape=pad_shape,
scale_factor=scale_factor,
flip=False)
]
return dict(img=[img], img_meta=[img_meta]) 示例2: _prepare_data
# 需要导入模块: from mmdet import datasets [as 别名]
# 或者: from mmdet.datasets import to_tensor [as 别名]
def _prepare_data(img, img_transform, cfg, device):
ori_shape = img.shape
img, img_shape, pad_shape, scale_factor = img_transform(
img, scale=cfg.data.test.img_scale)
img = to_tensor(img).to(device).unsqueeze(0)
img_meta = [
dict(
ori_shape=ori_shape,
img_shape=img_shape,
pad_shape=pad_shape,
scale_factor=scale_factor,
flip=False)
]
return dict(img=[img], img_meta=[img_meta])