/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

1 vote157 views1 answer
Python singleton
Python
0 vote150 views1 answer
Lsit append in python
Python
0 vote149 views1 answer
Add option in python script
Python
1 vote158 views2 answers
Python listen to keyboard input linux
Python
1 vote147 views1 answer
How to fill na python
Python
0 vote155 views2 answers
How to install modules to a specific version of python
Python
0 vote158 views3 answers
Polynomial regression python
Python
0 vote161 views1 answer
How to find in which directory my python code is running
Python
0 vote152 views3 answers
How to quickly draw a rectangle using Python's Turtle module.
Python
0 vote164 views2 answers
Python min value index from an array
Python