RuiTerrty/RemoteSensingChangeDetection-RSCD.HA2F
0
1a
2 �~0h� � @ sF d dl Z d dlZd dlmZ d dlZd dlZG dd� dejj j3�Z4dS )� N)�joinc @ s. e Zd Zd dd�Zdd� Zdd � Zd5d � ZdS )
�Dataset�data/�trainNc sf t �t� �d��| _� �fdd�| jD �| _� �fdd�| jD �| _� �fdd�| jD �| _|| _d S )N�Ac s g | ]}t � �d |��qS )r ��osp��.0�x�� file_root�mode� �:C:\Users\82607\Desktop\code\ChangeViT\.\dataset\dataset.py�6<listcomp> � z$Dataset.__init__.<locals>.<listcomp>c s g | ]}t � �d |��qS )�Br r r r r r
r c s g | ]}t � �d |��qS )�labelr r r r r r r )�os�listdirr � file_list�7pre_images� post_images�gts� transform)�selfr
r r r r r �__init__ s8 zDataset.__init__c C s9 t | j�S )N)�lenr )r r r r �__len__ s zDataset.__len__c C sn | j | }| j| }| j| }t�|�}t�|d�}t�|�}tj||fdd�}| jrf| �||�\}}||fS )Nr � )�axis)r r r �cv2�imread�numpy� concatenater ) r �idxZpre_image_name�10label_nameZpost_image_nameZ pre_imager Z11post_image�imgr r r � __getitem__ s 12131415 16zDataset.__getitem__c C s& t �| j| �}|jd |jd d�S )Nr � )�height�width)r" r# r �shape)r r& r( r r r � get_img_info% s zDataset.get_img_info)r r N)�__name__�17__module__� __qualname__r r r) r. r r r r r s 18 r ) r" r �os.pathr r r$ �torch.utils.data�torch�utils�datar r r r r �<module> s19