Android人臉識別Demo豎屏YUV方向調整和圖片保存

本博客包含三個經常使用方法,用於怒放Android版人臉識別Demo中豎屏使用時送入yuv數據,但一直沒法識別的狀況。 1.首先能夠嘗試順時針旋轉90°或270°,而後送入識別SDK。 2.旋轉方向後依然沒法識別時,能夠嘗試saveImg( ),保存本地檢查圖片是否符合要求。 | 在這裏插入圖片描述code

/**
     * 視頻順時針旋轉90
     * 該方法僅僅在豎屏時候使用
     * */
    public static byte[] rotateYUV420Degree90(byte[] data, int imageWidth,
                                        int imageHeight) {

        byte[] yuv = new byte[imageWidth * imageHeight * 3 / 2];
        int i = 0;
        for (int x = 0; x < imageWidth; x++) {
            for (int y = imageHeight - 1; y >= 0; y--) {
                yuv[i] = data[y * imageWidth + x];
                i++;
            }
        }
        i = imageWidth * imageHeight * 3 / 2 - 1;
        for (int x = imageWidth - 1; x > 0; x = x - 2) {
            for (int y = 0; y < imageHeight / 2; y++) {
                yuv[i] = data[(imageWidth * imageHeight) + (y * imageWidth) + x];
                i--;
                yuv[i] = data[(imageWidth * imageHeight) + (y * imageWidth)
                        + (x - 1)];
                i--;
            }
        }
        return yuv;
    }


    public static byte[] YUV420spRotate270(byte[] src, int width, int height) {
        int count = 0;
        int uvHeight = height >> 1;
        int imgSize = width * height;
        byte[] des = new byte[imgSize * 3 >> 1];
        //copy y
        for (int j = width - 1; j >= 0; j--) {
            for (int i = 0; i < height; i++) {
                des[count++] = src[width * i + j];
            }
        }
        //u,v
        for (int j = width - 1; j > 0; j -= 2) {
            for (int i = 0; i < uvHeight; i++) {
                des[count++] = src[imgSize + width * i + j - 1];
                des[count++] = src[imgSize + width * i + j];
            }
        }
        return des;
    }

 private int i = 1;
    private String path = Environment.getExternalStorageDirectory().getAbsolutePath() + "/0Face/";
    private Calendar now = new GregorianCalendar();
    private SimpleDateFormat simpleDate = new SimpleDateFormat("yyyyMMddHHmmss", Locale.getDefault());
    private String fileName = simpleDate.format(now.getTime());


    /**
     * @param data yuv圖像數據
     * @param width 
     * @param height
     */
    public void saveImg(byte[] data, int width, int height) {
        File dir = new File(path);
        if (!dir.exists()) dir.mkdirs();
        File f = new File(path + (fileName + "-" + i++) + ".jpg");
        FileOutputStream fOut = null;
        try {
            //yuv轉成bitmap
            YuvImage image = new YuvImage(data, ImageFormat.NV21, width, height, null);
            ByteArrayOutputStream stream = new ByteArrayOutputStream();
            image.compressToJpeg(new Rect(0, 0, width, height), 80, stream);
            Bitmap bmp = BitmapFactory.decodeByteArray(stream.toByteArray(), 0, stream.size());
           //bitmap保存至本地
            fOut = new FileOutputStream(f);
            bmp.compress(Bitmap.CompressFormat.JPEG, 100, fOut);
            fOut.flush();
            fOut.close();
            bmp.recycle();
            stream.close();
        } catch (Exception ex) {
            Log.e("Sys", "Error:" + ex.getMessage());
        }
    }
相關文章
相關標籤/搜索