[pytorch] tensor의 값을 뽑고싶을때

미남홀란드·2023년 2월 10일

yolo v8 리턴되는 바운딩 박스의 값을 출력하는 도중에 텐서때문에 많은 스트레스를 받았다.xyxy 값으로 리턴을 해주는데 처음보는 텐서의 값이었는데 dtype 을 해도 안먹어서 뭐지 하다가 별짓거리를 다하다가 발견해냈다.

#shape확인 tensor([[1351.50000, 123.50000, 245.00000, 33.00000]])

IndexError: invalid index of a 0-dim tensor. Use tensor.item() in Python or tensor.item<T>() in C++ to convert a 0-dim tensor to a

print('shape확인@@@@@@@@',x1.item() ,y1.item() ,x2.item() ,y2.item())

이렇게 해주면 텐서에 들어가있는 고유값을 뽑아준다. 그걸 numpy로 변환하던지 핸들링을 자유롭게 해주면 될거같다

profile
AI engineer

0개의 댓글