IGOS_pp/args.py at master · Deep-Machine-Vision/IGOS_pp · 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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
import argparse


def init_args():

    parser = argparse.ArgumentParser(
        description='Generate explanations using I-GOS and iGOS++.'
    )

    parser.add_argument(
        '--model',
        metavar='M',
        type=str,
        choices=['vgg19', 'resnet50', 'm-rcnn', 'f-rcnn', 'yolov3spp'],
        default='resnet50',
        help='The model to use for making predictions.')

    parser.add_argument(
        '--model_file',
        metavar='MF',
        default='./weight/maskrcnn_resnet50_fpn_coco-bf2d0c1e.pth',
        type=str,
        help='the path to the model weight file to be used.')

    parser.add_argument(
        '--data',
        metavar='D',
        type=str,
        required=True,
        help='the path to the data to be explained.')

    parser.add_argument(
        '--dataset',
        type=str,
        choices=['imagenet', 'coco', 'voc'],
        default='imagenet',
        help='The dataset to use for making predictions.')

    parser.add_argument(
        '--shuffle',
        action='store_true',
        help='Shuffle the dataset.')

    parser.add_argument(
        '--size',
        type=int,
        default=28,
        help='The resolution of mask to be generated.')

    parser.add_argument(
        '--input_size',
        type=int,
        default=224,
        help='The input size to the network.')

    parser.add_argument(
        '--num_samples',
        type=int,
        default=5000,
        help='The number of samples to run explanation on.')

    parser.add_argument(
        '--manual_seed',
        type=int,
        default=63,
        help='The manual seed for experiments.')

    parser.add_argument(
        '--method',
        required=True,
        type=str,
        choices=['I-GOS', 'iGOS+', 'iGOS++'],
        default='I-GOS'
    )

    parser.add_argument(
        '--opt',
        required=True,
        type=str,
        choices=['LS', 'NAG'],
        default='NAG',
        help='The optimization algorithm.'
    )

    parser.add_argument(
        '--diverse_k',
        type=int,
        default=2)

    parser.add_argument(
        '--init_posi',
        type=int,
        default=0,
        help='The initialization position, which cell of the K x K grid will be used to initialize the mask with nonzero values (use init_val to control it)')
    """
            If K = 2:      If K = 3:
            -------        ----------
            |0 |1 |        |0 |1 |2 |
            -------        ----------
            |2 |3 |        |3 |4 |5 |
            -------        ----------
                           |6 |7 |8 |
                           ----------
    """

    parser.add_argument(
        '--init_val',
        type=float,
        default=0.,
        help='The initialization value used to initialize the mask in only one cell of the K x K grid.')

    parser.add_argument(
        '--L1',
        type=float,
        default=1
    )

    parser.add_argument(
        '--L2',
        type=float,
        default=20
    )

    parser.add_argument(
        '--ig_iter',
        type=int,
        default=20)

    parser.add_argument(
        '--iterations',
        type=int,
        default=15
    )

    parser.add_argument(
        '--alpha',
        type=float,
        default=1000
    )

    return parser.parse_args()