predict.py 395 B

123456789101112131415161718192021
  1. import sys
  2. sys.path.append('/home/nvidia/newdisk/hkpc/software')
  3. from ultralytics import YOLO
  4. import time
  5. model = YOLO('/home/nvidia/newdisk/hkpc/ultralytics-main/best.pt')
  6. while 1:
  7. start_time = time.time()
  8. results = model('data/1.jpg')
  9. end_time = time.time()
  10. elapsed_time = end_time - start_time
  11. print(f"代码运行时间: {elapsed_time} 秒")
  12. time.sleep(0.5)