/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 vote131 views1 answer
Lineplot in plt
Python
0 vote149 views1 answer
Python How To Check Operating System
Python
0 vote140 views1 answer
Read a large dataframe in pandas
Python
0 vote170 views1 answer
Increase axis ticks pyplot
Python
1 vote157 views1 answer
Sklearn accuracy
Python
0 vote143 views1 answer
Required depend filed odoo
Python
8 votes133 views1 answer
Pygame draw rect syntax
Python
33 votes169 views5 answers
How to filter out all NaN values in pandas df
Python
0 vote136 views1 answer
Assert_series_equal
Python
0 vote129 views1 answer
Py2exe no console
Python