From a7ad5343f323bd7df99b4d7ead156c702c2853d4 Mon Sep 17 00:00:00 2001 From: Mehmet Durmaz <95125171+mdrmz@users.noreply.github.com> Date: Mon, 8 Jul 2024 10:11:01 +0300 Subject: [PATCH 1/2] Resnet Deneme --- anodet/feature_extraction.py | 6 ++++-- anodet/padim.py | 2 ++ anodet/utils.py | 8 ++++---- 3 files changed, 10 insertions(+), 6 deletions(-) diff --git a/anodet/feature_extraction.py b/anodet/feature_extraction.py index 4ec999e..8fd30ee 100644 --- a/anodet/feature_extraction.py +++ b/anodet/feature_extraction.py @@ -4,7 +4,7 @@ import torch import torch.nn.functional as F -from torchvision.models import resnet18, ResNet18_Weights, wide_resnet50_2, Wide_ResNet50_2_Weights +from torchvision.models import resnet18, ResNet18_Weights, wide_resnet50_2, Wide_ResNet50_2_Weights,resnet34,ResNet34_Weights from tqdm import tqdm from typing import List, Optional, Callable, cast from torch.utils.data import DataLoader @@ -29,12 +29,14 @@ def __init__(self, backbone_name: str, device: torch.device) -> None: """ super().__init__() - assert backbone_name in ['resnet18', 'wide_resnet50'] + assert backbone_name in ['resnet18', 'wide_resnet50','resnet34'] if backbone_name == 'resnet18': self.backbone = resnet18(weights=ResNet18_Weights.DEFAULT, progress=True) elif backbone_name == 'wide_resnet50': self.backbone = wide_resnet50_2(weights=Wide_ResNet50_2_Weights.DEFAULT, progress=True) + elif backbone_name == 'resnet34': + self.backbone = resnet34(weights=ResNet34_Weights.DEFAULT, progress=True) self.backbone.to(device) self.backbone.eval() diff --git a/anodet/padim.py b/anodet/padim.py index 437bab6..eb4a079 100644 --- a/anodet/padim.py +++ b/anodet/padim.py @@ -52,6 +52,8 @@ def __init__(self, backbone: str = 'resnet18', self.channel_indices = get_indices(100, 448, self.device) elif backbone == 'wide_resnet50': self.channel_indices = get_indices(550, 1792, self.device) + elif backbone == 'resnet34': + self.channel_indices = get_indices(100, 448, self.device) self.layer_indices = layer_indices if self.layer_indices is None: diff --git a/anodet/utils.py b/anodet/utils.py index 83a24b8..6eb8448 100644 --- a/anodet/utils.py +++ b/anodet/utils.py @@ -10,15 +10,15 @@ import os -standard_image_transform = T.Compose([T.Resize(224), - T.CenterCrop(224), +standard_image_transform = T.Compose([T.Resize(640), + T.CenterCrop(640), T.ToTensor(), T.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]) ]) -standard_mask_transform = T.Compose([T.Resize(224), - T.CenterCrop(224), +standard_mask_transform = T.Compose([T.Resize(640), + T.CenterCrop(640), T.ToTensor() ]) From 16eb26cc9a47d4d3480e6e599371866977b5113a Mon Sep 17 00:00:00 2001 From: Mehmet Durmaz <95125171+mdrmz@users.noreply.github.com> Date: Tue, 16 Jul 2024 00:31:56 +0300 Subject: [PATCH 2/2] Update utils.py --- anodet/utils.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/anodet/utils.py b/anodet/utils.py index 6eb8448..f3e2803 100644 --- a/anodet/utils.py +++ b/anodet/utils.py @@ -10,15 +10,15 @@ import os -standard_image_transform = T.Compose([T.Resize(640), - T.CenterCrop(640), +standard_image_transform = T.Compose([T.Resize(320), + T.CenterCrop(320), T.ToTensor(), T.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]) ]) -standard_mask_transform = T.Compose([T.Resize(640), - T.CenterCrop(640), +standard_mask_transform = T.Compose([T.Resize(320), + T.CenterCrop(320), T.ToTensor() ])