diff --git a/test_data/test_portrait_images/your_portrait_results/GalGadot.png b/test_data/test_portrait_images/your_portrait_results/GalGadot.png index 074441a..8aec7e7 100644 Binary files a/test_data/test_portrait_images/your_portrait_results/GalGadot.png and b/test_data/test_portrait_images/your_portrait_results/GalGadot.png differ diff --git a/test_data/test_portrait_images/your_portrait_results/guliNazha3.png b/test_data/test_portrait_images/your_portrait_results/guliNazha3.png index f230026..ae04f7f 100644 Binary files a/test_data/test_portrait_images/your_portrait_results/guliNazha3.png and b/test_data/test_portrait_images/your_portrait_results/guliNazha3.png differ diff --git a/test_data/test_portrait_images/your_portrait_results/kid1.png b/test_data/test_portrait_images/your_portrait_results/kid1.png index bbf8612..ae912e1 100644 Binary files a/test_data/test_portrait_images/your_portrait_results/kid1.png and b/test_data/test_portrait_images/your_portrait_results/kid1.png differ diff --git a/test_data/test_portrait_images/your_portrait_results/kid2.png b/test_data/test_portrait_images/your_portrait_results/kid2.png index 24c1a9c..56a1744 100644 Binary files a/test_data/test_portrait_images/your_portrait_results/kid2.png and b/test_data/test_portrait_images/your_portrait_results/kid2.png differ diff --git a/test_data/test_portrait_images/your_portrait_results/kid3.png b/test_data/test_portrait_images/your_portrait_results/kid3.png index c648af2..bc55994 100644 Binary files a/test_data/test_portrait_images/your_portrait_results/kid3.png and b/test_data/test_portrait_images/your_portrait_results/kid3.png differ diff --git a/test_data/test_portrait_images/your_portrait_results/man.png b/test_data/test_portrait_images/your_portrait_results/man.png index 182f22e..065e2c8 100644 Binary files a/test_data/test_portrait_images/your_portrait_results/man.png and b/test_data/test_portrait_images/your_portrait_results/man.png differ diff --git a/test_data/test_portrait_images/your_portrait_results/man2.png b/test_data/test_portrait_images/your_portrait_results/man2.png index 84132fc..a8c2d12 100644 Binary files a/test_data/test_portrait_images/your_portrait_results/man2.png and b/test_data/test_portrait_images/your_portrait_results/man2.png differ diff --git a/test_data/test_portrait_images/your_portrait_results/man4.png b/test_data/test_portrait_images/your_portrait_results/man4.png index 69d0468..1afe81a 100644 Binary files a/test_data/test_portrait_images/your_portrait_results/man4.png and b/test_data/test_portrait_images/your_portrait_results/man4.png differ diff --git a/test_data/test_portrait_images/your_portrait_results/man5.png b/test_data/test_portrait_images/your_portrait_results/man5.png index c3549ae..f0f7845 100644 Binary files a/test_data/test_portrait_images/your_portrait_results/man5.png and b/test_data/test_portrait_images/your_portrait_results/man5.png differ diff --git a/test_data/test_portrait_images/your_portrait_results/smile.png b/test_data/test_portrait_images/your_portrait_results/smile.png index a6caa1c..1946287 100644 Binary files a/test_data/test_portrait_images/your_portrait_results/smile.png and b/test_data/test_portrait_images/your_portrait_results/smile.png differ diff --git a/u2net_portrait_demo.py b/u2net_portrait_demo.py index 81d8feb..516272a 100644 --- a/u2net_portrait_demo.py +++ b/u2net_portrait_demo.py @@ -39,26 +39,34 @@ def crop_face(img, face): # crop the face with a bigger bbox hmw = h - w - hpad = int(h/2)+1 - wpad = int(w/2)+1 + # hpad = int(h/2)+1 + # wpad = int(w/2)+1 l,r,t,b = 0,0,0,0 - left = x-wpad + lpad = int(float(w)*0.4) + left = x-lpad if(left<0): + l = lpad-x left = 0 - l = wpad-x - right = x+w+wpad + + rpad = int(float(w)*0.4) + right = x+w+rpad if(right>width): - right = width r = right-width - top = y - hpad + right = width + + tpad = int(float(h)*0.6) + top = y - tpad if(top<0): + t = tpad-y top = 0 - t = hpad-y - bottom = y+h+int(hpad*0.5) + + bpad = int(float(h)*0.2) + bottom = y+h+bpad if(bottom>height): - bottom = height b = bottom-height + bottom = height + im_face = img[top:bottom,left:right] if(len(im_face.shape)==2):