/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

0 vote180 views1 answer
Waiting in python. time , sleep
Python
0 vote225 views1 answer
Selenium ways of finding
Python
0 vote176 views1 answer
Soustraire deux listes python
Python
0 vote231 views1 answer
How to make a python file that prints out a random element from a list
Python
0 vote233 views1 answer
How to preserve white space when joining an array python
Python
0 vote193 views1 answer
Python how to get current line number
Python
33 votes176 views6 answers
Merge dataframe
Python
0 vote183 views1 answer
How to count a consecutive series of positive or negative values in a column in python
Python
1 vote228 views1 answer
Django queryset first element
Python Django
12 votes187 views6 answers
Flask
Python