/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 votes270 views7 answers
Pyhton virtual env
Python
1 vote281 views1 answer
Streamlit button to load a file
Python
0 vote319 views1 answer
Pipenv installs
Python
34 votes295 views5 answers
Pandas max columns
Python
0 vote278 views1 answer
Python how to invert an array
Python
14 votes276 views2 answers
Python tim script
Python
3 votes298 views3 answers
Download playlist from youtube python
Python
10 votes285 views3 answers
Dataframein python
Python
3 votes289 views1 answer
Python list of random float numbers
Python
0 vote285 views1 answer
Change color of text button pyqt5
Python