/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

3 votes364 views1 answer
Pyautogui moveTo overtime
Python
2 votes338 views1 answer
Ipython.display clear_output
Python
0 vote352 views1 answer
Python convert exponential to int
Python
4 votes446 views3 answers
Python list add element to front
Python
0 vote330 views1 answer
MovieWriter stderr: ffmpeg: error while loading shared libraries: libopenh264.so.5: cannot open shared object file: No such file or directory
Python
0 vote341 views2 answers
Generate gif py
Python
23 votes520 views6 answers
Flask getting started
Python Flask
13 votes332 views2 answers
Ipykernel install
Python
0 vote369 views1 answer
Instabot python
Python
0 vote385 views1 answer
How to apply a function to multiple columns in pandas dataframe
Python