Mjolnir65/FasterRCNN
0
1B
2 �� h � @ s6 d dl mZmZ d dlZd dlmZ G dd� d�ZdS )� )�List�TupleN)�Tensorc @ s@ e Zd ZdZeeeeef dd�dd�Ze j3d d�dd�Z dS ) � ImageListaq 4 Structure that holds a list of images (of possibly5 varying sizes) as a single tensor.6 This works by padding the images to the same size,7 and storing in a field the original sizes of each image8 9 Args:10 tensors (tensor): Tensor containing images.11 image_sizes (list[tuple[int, int]]): List of Tuples each containing size of images.12 N)�tensors� image_sizes�returnc C s || _ || _d S )N)r r )�selfr r � r13 �8/home/rohit/Desktop/Sem6/CV/q1.2/detection/image_list.py�__init__ s zImageList.__init__)�devicer c C s | j �|�}t|| j�S )N)r �tor r )r r
� cast_tensorr14 r15 r r s z ImageList.to) �__name__�16__module__� __qualname__�__doc__r r r �intr �torchr
r r17 r18 r19 r r s 20 r )�typingr r r r r r21 r22 r23 r �<module> s