터미널에서 추론 결과 보기

libsixel 웹사이트의 이미지입니다.

동기 부여

원격 머신에 연결하면 일반적으로 이미지 결과를 시각화할 수 없거나 GUI를 사용하여 로컬 장치로 데이터를 이동해야 합니다. VSCode 통합 터미널을 사용하면 이미지를 직접 렌더링할 수 있습니다. 다음은 이를 다음과 함께 사용하는 방법에 대한 짧은 데모입니다. ultralytics 와 함께 예측 결과.

통합 터미널을 사용하여 이미지를 보기 위한 VSCode 호환 프로토콜은 다음과 같습니다. sixel 그리고 iTerm. 이 가이드에서는 sixel 프로토콜을 사용합니다.

프로세스

1. 먼저 설정을 활성화해야 합니다. terminal.integrated.enableImages 그리고 terminal.integrated.gpuAcceleration 를 입력합니다.

   "terminal.integrated.gpuAcceleration": "auto" # "auto" is default, can also use "on"
   "terminal.integrated.enableImages": false
   

1. 설치 python-sixel 라이브러리를 사용하세요. 이것은 포크 의 PySixel 라이브러리를 더 이상 유지 관리하지 않습니다.

   pip install sixel
   

2. 관련 라이브러리 가져오기

   import io
   
   import cv2 as cv
   
   from ultralytics import YOLO
   from sixel import SixelWriter
   

3. 모델을 로드하고 추론을 실행한 다음 결과를 플로팅하고 변수에 저장합니다. 예측 모드 페이지에서 추론 인수 및 결과 작업에 대해 자세히 알아보세요.

   from ultralytics import YOLO
   
   # Load a model
   model = YOLO("yolov8n.pt")
   
   # Run inference on an image
   results = model.predict(source="ultralytics/assets/bus.jpg")
   
   # Plot inference results
   plot = results[0].plot() #(1)!
   

   1. 사용할 수 있는 인수를 확인하려면 플롯 메서드 매개변수를 참조하세요.

4. 이제 OpenCV를 사용하여 numpy.ndarray 에 bytes 데이터를 입력합니다. 그런 다음 io.BytesIO 를 사용하여 "파일과 같은" 객체를 만듭니다.

   # Results image as bytes
   im_bytes = cv.imencode(
       ".png", #(1)!
       plot,
       )[1].tobytes() #(2)!
   
   # Image bytes as a file-like object
   mem_file = io.BytesIO(im_bytes)
   

   1. 다른 이미지 확장자도 사용할 수 있습니다.
   2. 인덱스에 있는 개체만 1 를 반환해야 합니다.

5. 만들기 SixelWriter 인스턴스를 사용한 다음 .draw() 메서드를 사용하여 터미널에 이미지를 그립니다.

   # Create sixel writer object
   w = SixelWriter()
   
   # Draw the sixel image in the terminal
   w.draw(mem_file)
   

추론 결과 예시

전체 코드 예제

import io

import cv2 as cv

from ultralytics import YOLO
from sixel import SixelWriter

# Load a model
model = YOLO("yolov8n.pt")

# Run inference on an image
results = model.predict(source="ultralytics/assets/bus.jpg")

# Plot inference results
plot = results[0].plot() #(3)!

# Results image as bytes
im_bytes = cv.imencode(
    ".png", #(1)!
    plot,
    )[1].tobytes() #(2)!

mem_file = io.BytesIO(im_bytes)
w = SixelWriter()
w.draw(mem_file)

1. 다른 이미지 확장자도 사용할 수 있습니다.
2. 인덱스에 있는 개체만 1 를 반환해야 합니다.
3. 사용할 수 있는 인수를 확인하려면 플롯 메서드 매개변수를 참조하세요.

댓글