brails.processors.nfloors_detector.lib.infer_detector module

class brails.processors.nfloors_detector.lib.infer_detector.Infer(verbose=1)

Bases: object

display(preds, imgs, imshow=True, imwrite=False)

load_model(model_path, classes_list, use_gpu=True)

predict(img_path, threshold=0.5)