/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

7 votes140 views2 answers
Pygame keyboard input
Python
6 votes149 views6 answers
What is tuple in python
Python
1 vote174 views1 answer
Correr servidor django
Python Django
38 votes173 views6 answers
Remove substring from string python
Python
10 votes147 views2 answers
Convert hex to decimal python
Python
4 votes138 views4 answers
Download pdf python
Python
0 vote164 views1 answer
Python rotate pdf pages
Python
-1 vote184 views1 answer
Regex find all sentences python
Python
2 votes136 views2 answers
Build image from dockerfile
Python
0 vote151 views2 answers
Python-wordpress-xmlrpc custom fields
Python