|
@@ -10,11 +10,9 @@ import torch
|
|
|
import base64
|
|
import base64
|
|
|
import random
|
|
import random
|
|
|
|
|
|
|
|
-# YOLO_DIR = '/workspace/yolov5'
|
|
|
|
|
-YOLO_DIR = '/Users/sxkj/opt/python-workspace/yili-ocr/yl-ocr-layout/yolov5'
|
|
|
|
|
|
|
+YOLO_DIR = '/workspace/yolov5'
|
|
|
# WEIGHTS = '/data/yolov5/runs/train/yolov5x_layout_reuslt37/weights/best.pt'
|
|
# WEIGHTS = '/data/yolov5/runs/train/yolov5x_layout_reuslt37/weights/best.pt'
|
|
|
-# WEIGHTS = '/workspace/best.pt'
|
|
|
|
|
-WEIGHTS = '/Users/sxkj/opt/python-workspace/yili-ocr/yl-ocr-layout/best.pt'
|
|
|
|
|
|
|
+WEIGHTS = '/workspace/best.pt'
|
|
|
|
|
|
|
|
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
|
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
|
|
|
|
|
