/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

5 votes132 views1 answer
Check django object exists
Python
0 vote134 views1 answer
Insertar en una lista anidada python
Python
3 votes124 views1 answer
Create or update django models
Python
3 votes131 views1 answer
How to concatenate a string with int in python
Python
2 votes121 views2 answers
Django search
Python
0 vote112 views1 answer
Plt tickpad
Python
0 vote98 views1 answer
Python3 delete file
Python
1 vote126 views1 answer
Np.multiply
Python
0 vote141 views1 answer
Redirect if not logged in django
Python
-1 vote153 views4 answers
Real python linear regression
Python