/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 votes112 views1 answer
How to rotate screen with python
Python
1 vote123 views1 answer
Db_index django
Python Django
3 votes123 views1 answer
2x + 9y = -7 6x - 3y = 9
Python
0 vote119 views1 answer
Find the middle of the document in the image opencv
Python
0 vote71 views1 answer
Dataframe pandas filter rows by list of values
Python
65 votes118 views4 answers
How to sum all the numbers in a list in python
Python
0 vote105 views1 answer
Python download video from instagram
Python
1 vote126 views1 answer
Python list add if not present
Python
1 vote108 views1 answer
Python find string in list
Python
5 votes115 views3 answers
Python calculate prime numbers until numer
Python