Use np.array to more efficiently create tensors

CellProfiling / HPA-Cell-Segmentation

Apache License 2.0

95 stars 26 forks source link

Use np.array to more efficiently create tensors #28

Closed cwinsnes closed 1 year ago

cwinsnes commented 1 year ago

According to warnings that arise when running the Cellsegmentator, one should supply torch.Tensor with a single ndarray rather than multiple in a list in order for the conversion to be more efficient.

This PR adds a quick call to np.array for the input to the Tensor call.