/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 vote238 views1 answer
Flask where to put db.create_all
Python Flask
1 vote216 views1 answer
Python previous answer
Python
2 votes186 views3 answers
How to unlist a list in python
Python
1 vote214 views1 answer
How to count down in python using turtle graphics
Python
1 vote177 views1 answer
Python datetime round to nearest hour
Python
9 votes255 views2 answers
Django model form import
Python Django
0 vote207 views1 answer
Parsing a file and adding a number at starting of every line sentence python
Python
0 vote229 views1 answer
Modules django
Python Django
0 vote213 views1 answer
Running code once in a while loop python
Python
0 vote198 views1 answer
Webhook logger python
Python