perf(util): 辅助函数get_deivce：返回gpu或cpu device

Author: zjZSTU

py/lib/utils/util.py

@@ -11,6 +11,10 @@
 import torch
 
 
+def get_device():
+    return torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+
+
 def iou(pred_box, target_box):
     """
     计算候选建议和标注边界框的IoU