new share

gistfile1.txt

def godspeed():
    runner = GeneralizedRCNNRunner()
    cfg = runner.get_default_cfg()
    cfg.merge_from_file(r"/home/pi/Desktop/train/mask_model2_config/config2 (2).yml")
    cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.9  # set threshold for this model
    cfg.MODEL.DEVICE = "cpu"
    cfg.QUANTIZATION.BACKEND = 'qnnpack'
    torch.backends.quantized.engine = 'qnnpack' 
    model = create_predictor('/home/pi/Desktop/train/predictor_path_mask2_custom')
    #mp.set_start_method("spawn",force = True)
    predictor = DemoPredictor(model)