U-2-Net icon indicating copy to clipboard operation
U-2-Net copied to clipboard

convert u2net_portrait.pth to coreml

Open mgstar1021 opened this issue 3 years ago • 2 comments

I convert u2net_portrait.pth to mlmodel and result is different. I can find there is a normalized code. pred = 1- d1[:,0,:,:]

How can I do using coremltools?

Origin img_1585 2

python model img_1585_

mlmodel out_p0

mgstar1021 avatar Jul 23 '21 23:07 mgstar1021

Can please share the CoreML model?

ioskevinshah avatar Feb 15 '22 14:02 ioskevinshah

@mgstar1021 can you please share the CoreML model? Thanks

adirkol avatar Mar 31 '22 10:03 adirkol