Default function

1. load_model
2. predict
3. predict_file
4. on_train_completed

load_model

Load the model and checkpoint here

Example:

class Model:
    ...

def load_model():
    model = Model()
    model.load_state_dict(torch.load(modelLoader.path))
    model.to(device)
    model.eval()
    modelLoader.update_model(model)

predict(data)

Predict using http post body

Example:

def predict(data):
    model = modelLoader.get_model()
    files = magic.download(data['images'])
    predict = []
    file_name = []
    for file in files:
        img, label = load_image(file)
        result = model(img)
        _, pred = torch.max(result, 1)
        predict.append(int(pred))
        file_name.append(label)
    result = {
        'data': {
            'predict': predict,
            'file_name': file_name
        }
    }
    return result

predict_file(files)

Predict using an http post file

Example:

def predict_file(files):
    model = modelLoader.get_model()
    files = magic.save(files)
    predict = []
    file_name = []
    for file in files:
        img, label = load_image(file)
        result = model(img)
        _, pred = torch.max(result, 1)
        predict.append(int(pred))
        file_name.append(label)
    result = {
        'data': {
            'predict': predict,
            'file_name': file_name
        }
    }
    return result

on_train_completed(metric, config)