/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 votes284 views2 answers
How to change colour of rows in csv using pandas
Python
0 vote284 views1 answer
Treat NaN as a category
Python
0 vote289 views1 answer
How to colour letters in python
Python
19 votes284 views2 answers
How to find the datatype of a dataframe in python
Python
0 vote278 views1 answer
Python download video from instagram
Python
8 votes300 views5 answers
Convert integer to binary in python
Python
6 votes280 views3 answers
Pd df to series
Python
15 votes270 views3 answers
Remove punctuation python string library
Python
0 vote274 views1 answer
Find and replace subword in word python regex
Python
29 votes274 views3 answers
For x in range(1,100, 10): print(x)
Python