Fix device mismatch: maskrcnn detector lands on cuda under mmgps lingering default device, but torch.from_numpy() input stays on cpu

lib/pymaf/utils/imutils.py

@@ -115,8 +115,13 @@ def process_image(img_file,
     # detection for bbox
     detector = detection.maskrcnn_resnet50_fpn(pretrained=True)
     detector.eval()
+    # Under mmgp'''s lingering torch.set_default_device('''cuda'''), this freshly-constructed
+    # detector'''s weights land on GPU automatically, but torch.from_numpy() always returns
+    # a CPU tensor (it shares memory with the numpy buffer). Match the input to wherever
+    # the detector actually ended up instead of assuming either device.
+    detector_device = next(detector.parameters()).device
     predictions = detector(
-        [torch.from_numpy(img_for_crop).permute(2, 0, 1) / 255.])[0]
+        [(torch.from_numpy(img_for_crop).permute(2, 0, 1) / 255.).to(detector_device)])[0]
     human_ids = torch.logical_and(
         predictions["labels"] == 1,
         predictions["scores"] == predictions["scores"].max()).nonzero().squeeze(1)