Camilo Gonçalves
Results
1
comments of
Camilo Gonçalves
@Jaftem try changing line 50 on run_ssd_live_demo.py from this: `predictor = create_mobilenetv2_ssd_lite_predictor(net, candidate_size=200)` to this: `predictor = create_mobilenetv2_ssd_lite_predictor(net, candidate_size=200, device=torch.device('cuda'))` Initializing the Predictor class this way solves this issue without...