File size: 1,225 Bytes
998bb30 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
from PIL import Image
import numpy as np
import cv2
import matplotlib.pyplot as plt
import os
from attacks.CleanSheet.models.resnet import resnet34, resnet18
from attacks.CleanSheet.models.vgg import vgg16
from attacks.CleanSheet.models.mobilenet_v2 import mobilenet_v2
import torch.nn as nn
from attacks.CleanSheet.utils_ import Trigger
import torchvision
from torchvision import transforms
from attacks.CleanSheet.poison_dataset import PoisonDataset
import numpy as np
from PIL import Image
from torch.nn import functional as F
from pytorch_grad_cam import (
GradCAM,
HiResCAM,
ScoreCAM,
GradCAMPlusPlus,
AblationCAM,
XGradCAM,
EigenCAM,
FullGrad,
)
from pytorch_grad_cam.utils.model_targets import ClassifierOutputTarget
from pytorch_grad_cam.utils.image import show_cam_on_image
from torch.utils.data import DataLoader
import torch.optim as optim
from tqdm import tqdm
from torch.optim import lr_scheduler
import os
import torch
import torch.nn as nn
from attacks.CleanSheet.utils_ import Trigger
import torchvision
from torchvision import transforms
from attacks.CleanSheet.poison_dataset import PoisonDataset
import numpy as np
from torch.nn import functional as F
from PIL import Image
|