Skip to content

Commit fc87614

Browse files
committed
remove device_id setting in online predict
1 parent 116180f commit fc87614

File tree

3 files changed

+0
-3
lines changed

3 files changed

+0
-3
lines changed

tools/infer/text/predict_cls.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -248,7 +248,6 @@ def save_cls_res(cls_res_all, img_paths, include_score=False, save_path="./cls_r
248248
# img_paths = img_paths[:250]
249249

250250
ms.set_context(mode=args.mode)
251-
ms.set_context(device_id=7)
252251

253252
# init classifier
254253
classifier = DirectionClassifier(args)

tools/infer/text/predict_rec.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -266,7 +266,6 @@ def save_rec_res(rec_res_all, img_paths, include_score=False, save_path="./rec_r
266266
# img_paths = img_paths[:250]
267267

268268
ms.set_context(mode=args.mode)
269-
ms.set_context(device_id=6)
270269

271270
# init detector
272271
text_recognize = TextRecognizer(args)

tools/infer/text/predict_system.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -174,7 +174,6 @@ def main():
174174
# img_paths = img_paths[:10]
175175

176176
ms.set_context(mode=args.mode)
177-
ms.set_context(device_id=5)
178177

179178
# init text system with detector and recognizer
180179
text_spot = TextSystem(args)

0 commit comments

Comments
 (0)