/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

2 votes101 views1 answer
Import validation error in django
Python
0 vote107 views2 answers
Append two 1d arrays python
Python
5 votes112 views1 answer
Python start simplehttpserver
Python
0 vote102 views1 answer
Query a PSQL Database From Python
Python
0 vote108 views1 answer
Jupyter notebook not opening
Python
0 vote65 views1 answer
Get size of square matrix python
Python
8 votes105 views6 answers
Declare variable in python
Python
3 votes111 views3 answers
AttributeError: 'Timedelta' object has no attribute 'minutes'
Python
-1 vote123 views1 answer
Change tuple in python
Python
0 vote103 views1 answer
Filter a pandas dataframe by length of list in a column
Python