ShapeDefense/config.py at master · aliborji/ShapeDefense · GitHub

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
44
45
from lib import *
import edge_detector


torch.manual_seed(42)
np.random.seed(42)
random.seed(42)


# This specifies whether edge detection should be done only on RGB (when FALSE) or use
# False means only rgb is used for edge detection
EDGE_ALL_CHANNELS = False

# edge_detect = None

edge_detect = edge_detector.detect_edge_gtsrb

# edge_detect = edge_detector.detect_edge_mnist

# edge_detect = edge_detector.detect_edge_new

# edge_detect = edge_detector.detect_edge_tiny

# edge_detect = edge_detector.compute_energy_matrix
# edge_detect = edge_detector.detect_edge_fashionmnist
# edge_detect = edge_detector.detect_edge_mnist

# edge_detect = edge_detector.detect_edge_sobel


# edge_detect = edge_detector.detect_edge_new_cifar


# edge detectors used: (for replication)
# detect_edge_new for DogVsCat  canny   !
# detect_edge_mnist for mnist and fashionMNIST
# compute_energy_matrix for dogs breeds
# detect_edge_gtsrb for gtsrb and icons, sketch
# cifar detect_edge_new_cifar OR detect_edge_tiny
# imanette detect_edge_tiny
# tiny imagenet detect_edge_tiny