Python datasets.to_tensor方法代碼示例

本文整理匯總了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])

開發者ID:dingjiansw101，項目名稱:AerialDetection，代碼行數:18，代碼來源:inference.py

示例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])

開發者ID:chanyn，項目名稱:Reasoning-RCNN，代碼行數:16，代碼來源:inference.py

注：本文中的mmdet.datasets.to_tensor方法示例由純淨天空整理自Github/MSDocs等開源代碼及文檔管理平台，相關代碼片段篩選自各路編程大神貢獻的開源項目，源碼版權歸原作者所有，傳播和使用請參考對應項目的License；未經允許，請勿轉載。