Commit 0c85f610 authored by Shaoshuai Shi's avatar Shaoshuai Shi
Browse files

conduct image normalize directly

parent 0743656b
...@@ -6,7 +6,6 @@ import torch ...@@ -6,7 +6,6 @@ import torch
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
import torchvision import torchvision
from kornia.enhance.normalize import normalize
class SegTemplate(nn.Module): class SegTemplate(nn.Module):
...@@ -112,7 +111,7 @@ class SegTemplate(nn.Module): ...@@ -112,7 +111,7 @@ class SegTemplate(nn.Module):
# Preprocess images # Preprocess images
if self.pretrained: if self.pretrained:
images = normalize(images, mean=self.norm_mean, std=self.norm_std) images = (images - self.norm_mean[None, :, None, None].type_as(images)) / self.norm_std[None, :, None, None].type_as(images)
x = images.cuda() x = images.cuda()
# Extract features # Extract features
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment