/Python/Pytorch dataloader to numpy array

Pytorch dataloader to numpy array

# index is the specific data your want in dataloader
import numpy as np
def transfer_dataloader(dataloader, index=0):
    cache_list = list(iter(dataloader))
    assert len(cache_list) > 0
    assert index < len(cache_list[0])
    result_list = np.array(list(map(lambda x: x[index].numpy(), cache_list)))
    return result_list

Similar codes

6 votes194 views1 answer
Getting dummies and input them to pandas dataframe
Python
8 votes197 views6 answers
Python tutorial
Python
13 votes203 views6 answers
Python length
Python
1 vote185 views1 answer
Robot framework log from python
Python
0 vote216 views1 answer
Pandas transform date format?
Python
51 votes176 views4 answers
Get date from datetime python
Python
0 vote247 views2 answers
Url namespaces for django rest router urls
Python Django
43 votes179 views1 answer
Abc list
Python
0 vote171 views1 answer
Pyttx3
Python
0 vote194 views1 answer
Import nifti to numpy
Python