/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

5 votes183 views3 answers
Ros python publisher
Python
32 votes158 views1 answer
How to add a column to a pandas df
Python
7 votes156 views2 answers
What does // mean in python
Python
0 vote162 views1 answer
How to find permutation of numbers in python
Python
0 vote198 views1 answer
Search object in array python
Python
0 vote188 views3 answers
How to list all the string's characters in python
Python
5 votes225 views6 answers
Youtube'
Python
1 vote167 views1 answer
Combination python
Python
1 vote207 views2 answers
Django class based views
Python Django
0 vote166 views2 answers
Tkinter change ttk button color
Python