Active-Learning/main.py at master · jpmrs1313/Active-Learning · 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
from query import (
    ClusterBasedSampling,
    RandomSampling,
    UncertaintySampling,
    RepresentativeSampling,
    UncertaintyWithClusteringSampling,
    RepresentativeWithClusteringSampling,
    HighestEntropyClusteringSampling,
    UncertaintyWithRepresentativeSampling,
    HighestEntropyUncertaintySampling,
    OutlierSampling,
    OutliersWithRepresentativeSampling,
    UncertaintyWithModelOutliersSampling
)
from tensorflow.keras.preprocessing.image import ImageDataGenerator
from active_learner import ActiveLearner
import os
import json
from estimators import *

os.environ["TF_FORCE_GPU_A_LLOW_GROWTH"] = "true"

# read file
with open("especificacoes.json", "r") as myfile:
    data = myfile.read()
# parse file
obj = json.loads(data)

datagen = ImageDataGenerator(rescale=1.0 / 255)

train_generator = datagen.flow_from_directory(
    str(obj["training_path"]),
    target_size=(128, 128),
    batch_size=200,
    class_mode="binary",
    shuffle=True,
)
test_generator = datagen.flow_from_directory(
    str(obj["testing_path"]),
    target_size=(128, 128),
    batch_size=500,
    class_mode="binary",
    shuffle=True,
)
validation_generator = datagen.flow_from_directory(
    str(obj["validation_path"]),
    target_size=(128, 128),
    batch_size=200,
    class_mode="binary",
    shuffle=True,
)
unlabeled_generator = datagen.flow_from_directory(
    str(obj["unlabeled_path"]),
    target_size=(128, 128),
    batch_size=500,
    class_mode="binary",
    shuffle=True,
)

X_initial, y_initial = next(train_generator)
X_test, y_test = next(test_generator)
X_validation, y_validation = next(validation_generator)
X_unlabeled, y_unlabeled = next(unlabeled_generator)


learner = ActiveLearner(
    locals()[obj["build_fn"]],
    UncertaintySampling(int(obj["n_instances"])),
    X_initial,
    y_initial
)

learner.estimator.model.load_weights("model.h5")

accuracy_values = learner.loop(
    X_test,
    y_test,
    X_unlabeled,
    float(obj["accuracy"]),
)