Get Pytorch - tensor values as a integer in python

Question

I have my output of my torch tensor which looks like below

(coordinate of a bounding box in object detection)

[tensor(299., device='cuda:0'), tensor(272., device='cuda:0'), tensor(327., device='cuda:0'), tensor(350., device='cuda:0')]

I wanted to extract each of the tensor value as an int in the form of minx,miny,maxx,maxy
so that I can pass it to a shapely function in the below form

from shapely.geometry import box
minx,miny,maxx,maxy=1,2,3,4
b = box(minx,miny,maxx,maxy)

What's the best way to do it? by avoiding, Cuda enabled or not or other exceptions?

Does this answer your question? How do I get value of a tensor in PyTorch? — End genocide - save Gaza
– End genocide - save Gaza, Commented Mar 30, 2021 at 21:54

Mustafa Aydın · Accepted Answer · 2021-03-30 16:25:32Z

2

minx, miny, maxx, maxy = [int(t.item()) for t in tensors]

where tensors is the list of tensors.

answered Mar 30, 2021 at 16:25

Mustafa Aydın

18.4k4 gold badges21 silver badges43 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Get Pytorch - tensor values as a integer in python

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related