mirror of
https://github.com/aladdinpersson/Machine-Learning-Collection.git
synced 2026-02-21 11:18:01 +00:00
20 lines
1.5 KiB
Plaintext
20 lines
1.5 KiB
Plaintext
|
U
|
|||
|
|
N<A` <00> @ s<> d dl Zd dlZd dlZd dlmZ d dlmZmZ d dl m
|
|||
|
|
Z
|
|||
|
|
G dd<06> de<07>Z e dkr<>e d<08>Z
ee
d d
|
|||
|
|
<EFBFBD>ZeD ]6\ZZeej<12> e
|
|||
|
|
ed <0B> e
|
|||
|
|
ed <0C> d dlZe<13><14> qpdS )
<0A> N)<01>Image)<02>Dataset<65>
|
|||
|
|
DataLoader)<01>
|
|||
|
|
save_imagec @ s$ e Zd Zdd<02> Zdd<04> Zdd<06> ZdS )<08>
|
|||
|
|
MapDatasetc C s || _ t<01>| j <00>| _d S <00>N)<04>root_dir<69>os<6F>listdir<69>
|
|||
|
|
list_files)<02>selfr <00> r
<00>AF:\Machine-Learning-Collection\ML\Pytorch\GANs\Pix2Pix\dataset.py<70>__init__
|
|||
|
|
s zMapDataset.__init__c C s
|
|||
|
|
t | j<01>S r )<02>lenr )r r
r
r <00>__len__ s zMapDataset.__len__c C s<> | j | }tj<02>| j|<02>}t<05>t<07>|<03><01>}|d d <00>d d<01>d d <00>f }|d d <00>dd <00>d d <00>f }t j
|
|||
|
|
||d<02>}|d }|d }t j |d<05>d }t j |d<05>d }||fS )NiX )<02>image<67>image0r r )r )
r r <00>path<74>joinr <00>np<6E>arrayr <00>open<65>config<69>both_transform<72>transform_only_input<75>transform_only_mask)r <00>index<65>img_fileZimg_pathr <00> input_image<67> target_image<67>
augmentationsr
r
r <00> __getitem__ s
|
|||
|
|
zMapDataset.__getitem__N)<06>__name__<5F>
|
|||
|
|
__module__<EFBFBD> __qualname__r r r" r
r
r
r r s r <00>__main__z data/train/<2F> )<01>
|
|||
|
|
batch_sizezx.pngzy.png)<15>numpyr r r <00>PILr <00>torch.utils.datar r <00>torchvision.utilsr r r# <00>dataset<65>loader<65>x<>y<>print<6E>shape<70>sys<79>exitr
r
r
r <00><module> s
|
|||
|
|
|
|||
|
|
|
|||
|
|
|