捷徑

regnet_x_800mf

torchvision.models.regnet_x_800mf(*, weights: Optional[RegNet_X_800MF_Weights] = None, progress: bool = True, **kwargs: Any) RegNet[原始碼]

設計網路設計空間 建構 RegNetX_800MF 架構。

參數:
  • weights (RegNet_X_800MF_Weights, optional) – 要使用的預訓練權重。請參閱下方的 RegNet_X_800MF_Weights 以取得更多詳細資訊和可能的值。依預設,不使用預訓練權重。

  • progress (bool, optional) – 若為 True,則將下載進度列顯示至 stderr。預設值為 True。

  • **kwargs – 傳遞至 torchvision.models.regnet.RegNettorchvision.models.regnet.BlockParams 類別的參數。關於類別的更多詳細資訊,請參閱原始碼

class torchvision.models.RegNet_X_800MF_Weights(value)[原始碼]

上述模型建構器接受以下值作為 weights 參數。RegNet_X_800MF_Weights.DEFAULT 等同於 RegNet_X_800MF_Weights.IMAGENET1K_V2。您也可以使用字串,例如 weights='DEFAULT'weights='IMAGENET1K_V1'

RegNet_X_800MF_Weights.IMAGENET1K_V1:

這些權重使用簡單的訓練配方,密切重現論文的結果。

acc@1 (在 ImageNet-1K 上)

75.212

acc@5 (在 ImageNet-1K 上)

92.348

min_size

height=1, width=1

categories

tench, goldfish, great white shark, … (已省略 997 個)

num_params

7259656

recipe

link

GFLOPS

0.80

檔案大小

27.9 MB

推論轉換可在 RegNet_X_800MF_Weights.IMAGENET1K_V1.transforms 中取得,並執行以下預先處理作業:接受 PIL.Image、批次 (B, C, H, W) 和單一 (C, H, W) 影像 torch.Tensor 物件。影像會使用 interpolation=InterpolationMode.BILINEAR 調整大小為 resize_size=[256],然後進行 crop_size=[224] 的中心裁剪。最後,值會先重新縮放至 [0.0, 1.0],然後使用 mean=[0.485, 0.456, 0.406]std=[0.229, 0.224, 0.225] 進行正規化。

RegNet_X_800MF_Weights.IMAGENET1K_V2:

這些權重透過使用 TorchVision 新訓練配方 的修改版本,改進原始論文的結果。也可作為 RegNet_X_800MF_Weights.DEFAULT 使用。

acc@1 (在 ImageNet-1K 上)

77.522

acc@5 (在 ImageNet-1K 上)

93.826

min_size

height=1, width=1

categories

tench, goldfish, great white shark, … (已省略 997 個)

num_params

7259656

recipe

link

GFLOPS

0.80

檔案大小

27.9 MB

推論轉換可在 RegNet_X_800MF_Weights.IMAGENET1K_V2.transforms 中取得,並執行以下預先處理作業:接受 PIL.Image、批次 (B, C, H, W) 和單一 (C, H, W) 影像 torch.Tensor 物件。影像會使用 interpolation=InterpolationMode.BILINEAR 調整大小為 resize_size=[232],然後進行 crop_size=[224] 的中心裁剪。最後,值會先重新縮放至 [0.0, 1.0],然後使用 mean=[0.485, 0.456, 0.406]std=[0.229, 0.224, 0.225] 進行正規化。

文件

取得 PyTorch 的完整開發者文件

檢視文件

教學

取得適用於初學者和進階開發者的深入教學

檢視教學

資源

尋找開發資源並獲得問題解答

檢視資源