In infer_simple.py the mask RCNN model is created as follows:
maskRCNN = mynn.DataParallel(maskRCNN, cpu_keywords=['im_info', 'roidb'],
minibatch=True, device_ids=[0]) # only support single GPU
I've noticed that the DataParallel module only supports a single GPU. I actually wanted to speed up my inference time i.e. time taken to visualise the detections of the model on an image since I have upward of 1000 images. Is there a way to visualise the detections on multiple image using multiple gpus? Also why can't the model support multiple GPUs?
In infer_simple.py the mask RCNN model is created as follows:
I've noticed that the DataParallel module only supports a single GPU. I actually wanted to speed up my inference time i.e. time taken to visualise the detections of the model on an image since I have upward of 1000 images. Is there a way to visualise the detections on multiple image using multiple gpus? Also why can't the model support multiple GPUs?