def test_extract_features_for_train_batch(self):
c = Params()
c.load_config_file(JSON_PATH)
# "inception_v3", "densenet121", "densenet169", "densenet201", "resnet50", "inception_resnet_v2",
# "vgg16", "mobilenet_v2"
MODEL_NAME_set = ["densenet169"]
SAMPLE_FIlENAME_set = [
"T_NC_2000_256", "T_NC_4000_256"
] # "T_NC_500_128", , "T_NC_4000_256" "T_NC_2000_256"
PATCH_TYPE_set = ["2000_256",
"4000_256"] # "500_128", , "4000_256" "2000_256"
type_set = zip(PATCH_TYPE_set, SAMPLE_FIlENAME_set)
for p_item, f_item in type_set:
for net_item in MODEL_NAME_set:
cnn = Transfer(c, net_item, p_item)
cnn.extract_features_for_train(f_item, 100)
def test_extract_features_for_train(self):
c = Params()
c.load_config_file(JSON_PATH)
cnn = Transfer(c, MODEL_NAME, PATCH_TYPE)
cnn.extract_features_for_train(SAMPLE_FIlENAME, 100)
