S3DIP/stopping_criterion.py at master · IAPP-Group/S3DIP · 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
""" Early stopping criterion for Deep Image Prior """

class EarlyStop:
    def __init__(self, size, patience):
        self.patience = patience
        self.wait_count = 0
        self.best_score = float('inf')
        self.best_epoch = 0
        self.img_collection = []
        self.stop = False
        self.size = size

    def check_stop(self, current, cur_epoch):
        # stop when variance doesn't decrease for consecutive P(patience) times
        best_updated = False  # added
        if current < self.best_score:
            self.best_score = current
            self.best_epoch = cur_epoch
            self.wait_count = 0
            should_stop = False
            best_updated = True
        else:
            self.wait_count += 1
            should_stop = self.wait_count >= self.patience
        return should_stop, best_updated

    def update_img_collection(self, cur_img):
        self.img_collection.append(cur_img)
        if len(self.img_collection) > self.size:
            self.img_collection.pop(0)

    def get_img_collection(self):
        return self.img_collection


def myMetric(x1, x2):
    return ((x1 - x2) ** 2).sum() / x1.size