Object Detection and Bounding Boxes

English Version pytorch
1

https://d2l.ai/chapter_computer-vision/bounding-box.html

2 Likes
2

" The input argument boxes can be either a tensor of length 4, or a two-dimensional tensor of shape (n, 4), where n is the number of bounding boxes."
However, according to the code, the argument boxes can only be a 2-dim tensor of shape(n, 4), since we are using index as [:, 0].