data preproccessing : mask = to_tensor
raidanis opened this issue · comments
raidanis commented
my masks shape is (512, 512), when i start the fitting i get an error of miss matching axes , i think the problem is here :
**def to_tensor(x, **kwargs):
return x.transpose(2, 0, 1).astype('float32')**
the function expect input of 3 dimentions not 2 , any one can help ??
github-actions commented
This issue is stale because it has been open 60 days with no activity. Remove stale label or comment or this will be closed in 7 days.
Pavel Iakubovskii commented
Hi, you can just expand the last dim, x[..., None]
raidanis commented
I fix it but , I'm facing problems in deployment with flask , can you help
sir?
…On Thu, May 9, 2024, 10:18 Pavel Iakubovskii ***@***.***> wrote:
Hi, you can just expand the last dim, x[..., None]
—
Reply to this email directly, view it on GitHub
<#859 (comment)>,
or unsubscribe
<https://github.com/notifications/unsubscribe-auth/ANFN64DNKGIJ26EU4IGW77DZBM5PTAVCNFSM6AAAAABEOCDKFKVHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMZDCMBSGI4DCNBZGQ>
.
You are receiving this because you are subscribed to this thread.Message
ID: ***@***.***>
Pavel Iakubovskii commented
You can ask a question, if I have an answer I will help :)