我目前正在使用 Keras 进行卫星图像分类,但在使用 Predict 和 Predict_generator 获得正确的预测时遇到了麻烦。
下面是我的代码
import os
import numpy as np
import pandas as pd
from keras.optimizers import Adam, SGD
from tools import load_val_datas, load_test_datas, make_predictions, make_submissions
from keras_tools import save_model, load_model
from callbacks import CustomCallbacks
from data_generator import ImageDataGenerator
from model import base_cnn
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
TRAIN_SIZE, VAL_SIZE, TEST_SIZE, TEST_SIZE_ADD = 30000, 10479, 40669, 20522
IMAGE_FIRST_DIM, N_COLORS = 32, 3
IMAGE_SIZE = IMAGE_FIRST_DIM * IMAGE_FIRST_DIM * N_COLORS
LABEL_SIZE = 17
DROPOUT = 0.25
BATCH_SIZE = 96 …Run Code Online (Sandbox Code Playgroud)