Solved by JosueCom in post #3 
              
              
                You can also convert each image before it goes to the array to a tensor via imgs.append(torch.from_numpy(img)), then use torch.stack(imgs) to turn the array into a tensor.