hzt_asc 1.1.0 → 1.1.1

package/ask.cc

@@ -152,6 +152,10 @@ Napi::Value ASK::faceDetectUrl(const Napi::CallbackInfo& info) {
     string filePath = info[0].As<Napi::String>().ToString();
     int Width, Height;
     MUInt8* imageData = processFileUrl(filePath.c_str(), Width, Height);
+    if (!imageData) {
+        Napi::TypeError::New(env, "image download error").ThrowAsJavaScriptException();
+        return errorData(env, "image download error"); //Napi::Number::New(info.Env(), -1);
+    }
     ASVLOFFSCREEN offscreen = { 0 };
     ColorSpaceConversion(Width, Height, ASVL_PAF_NV21, imageData, offscreen);
     ASF_MultiFaceInfo detectedFaces = { 0 };
@@ -244,6 +248,10 @@ Napi::Value ASK::faceFeatureExtractUrl(const Napi::CallbackInfo& info) {
    
     int Width, Height;
     MUInt8* imageData = processFileUrl(filePath.c_str(), Width, Height);
+    if (!imageData) {
+        Napi::TypeError::New(env, "image download error").ThrowAsJavaScriptException();
+        return errorData(env, "image download error"); //Napi::Number::New(info.Env(), -1);
+    }
     ASVLOFFSCREEN offscreen = { 0 };
     ColorSpaceConversion(Width, Height, ASVL_PAF_NV21, imageData, offscreen);
 
@@ -562,6 +570,10 @@ Napi::Value ASK::ImageFaceCompareUrl2(const Napi::CallbackInfo& info) {
     string filePath = info[0].As<Napi::String>().ToString();
     int Width, Height;
     MUInt8* imageData = processFileUrl(filePath.c_str(), Width, Height);
+    if (!imageData) {
+        Napi::TypeError::New(env, "image download error").ThrowAsJavaScriptException();
+        return errorData(env, "image download error"); //Napi::Number::New(info.Env(), -1);
+    }
     ASVLOFFSCREEN offscreen = { 0 };
     ColorSpaceConversion(Width, Height, ASVL_PAF_NV21, imageData, offscreen);
     
@@ -589,6 +601,10 @@ Napi::Value ASK::ImageFaceCompareUrl(const Napi::CallbackInfo& info) {
     string filePath = info[0].As<Napi::String>().ToString();
     int Width, Height;
     MUInt8* imageData = processFileUrl(filePath.c_str(), Width, Height);
+    if (!imageData) {
+        Napi::TypeError::New(env, "image download error").ThrowAsJavaScriptException();
+        return errorData(env, "image download error"); //Napi::Number::New(info.Env(), -1);
+    }
     ASVLOFFSCREEN offscreen = { 0 };
     ColorSpaceConversion(Width, Height, ASVL_PAF_NV21, imageData, offscreen);
 
@@ -762,7 +778,7 @@ size_t write_data(char *ptr, size_t size, size_t nmemb, void *userdata)
 //function to retrieve the image as cv::Mat data type
 cv::Mat curlImg(const char *img_url, int timeout=100000)
 {
-    cout<<img_url<<endl;
+    //cout<<img_url<<endl;
     vector<uchar> stream;
     CURL *curl = curl_easy_init();
     curl_easy_setopt(curl, CURLOPT_URL, img_url); //the img url
@@ -771,8 +787,14 @@ cv::Mat curlImg(const char *img_url, int timeout=100000)
     curl_easy_setopt(curl, CURLOPT_TIMEOUT, timeout); // timeout if curl_easy hangs, 
     curl_easy_setopt(curl, CURLOPT_FOLLOWLOCATION, 1); //重定向一次
     CURLcode res = curl_easy_perform(curl); // start curl
+    
     curl_easy_cleanup(curl); // cleanup
-    cout<<res<<" xxxx " << stream.size() << " " << stream.max_size()<<endl;
+    if (res != CURLE_OK) {
+        cout << "curl errcode=" << res << img_url <<endl;
+        cv::Mat dest(0, 0, 1);
+        return dest;
+    }
+    cout<<res<<" xxxx " << stream.size() << img_url << stream.max_size()<<endl;
     return imdecode(stream, -1); // 'keep-as-is'
 }
 
@@ -792,6 +814,10 @@ void modifyImageFormat(cv::Mat& out, cv::Mat& in, int channel)
 MUInt8* ASK::processFileUrl(const char*  url, int& picWidth, int& picHeight) {
     cv::Mat src = curlImg(url);
     int Width = src.cols;
+    if (Width == 0) {
+        src.release();
+        return NULL;
+    }
 	int Height = src.rows;
     int depth = src.depth();
     int channel = src.channels();
@@ -799,12 +825,13 @@ MUInt8* ASK::processFileUrl(const char*  url, int& picWidth, int& picHeight) {
     cout << "dims: " << src.dims << endl;
     cout << "channels: " << channel << endl;
     cout << "depth: " << depth << endl;
-
+    //src.reshape(channels, rows)
+    //src.assignTo(mid, depth)
     int wScore = Width % 4;
     int hScore = Height % 2;
     cv::Mat dest;
     if (wScore != 0 || hScore != 0) {
-        cout << "processFileUrl ou:" <<endl;
+        //cout << "processFileUrl ou:" <<endl;
         Width -= wScore;
         Height -= hScore;
         cv::Mat dst;
@@ -820,7 +847,7 @@ MUInt8* ASK::processFileUrl(const char*  url, int& picWidth, int& picHeight) {
 
         dst.release();
     } else {
-        cout << "processFileUrl ji:" <<endl;
+        //cout << "processFileUrl ji:" <<endl;
         if (depth != 0) { //bit数仅支持8
             cv::Mat mid;
             modifyImageFormat(mid, src, channel);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "hzt_asc",
-  "version": "1.1.0",
+  "version": "1.1.1",
   "main": "index.js",
   "description": "hzt_asc_v3.",
   "author": {

package/sample/demo.js

@@ -17,14 +17,26 @@ let start = Date.now();
 let ask = new addon.ASK(APPID, SDKKEY, ACTIVEKEY);
 //let ask2 = new addon.ASK(APPID, SDKKEY, ACTIVEKEY);
 printMem()
-let xx2 = ask.FaceDetectUrl("http://image.haizitong.com/cc123be3667143e3af5c53a5906c6b42");
-console.log("init end", xx2, Date.now() - start);
+// let xx2 = ask.FaceDetectUrl("http://image.haizitong.com/cc123be3667143e3af5c53a5906c6b42");
+// console.log("init end", xx2, Date.now() - start);
 
-let xx = ask.FaceDetectUrl("http://upload-file.haizitong.com/api/getFile/119c44a5-193f-4953-b59c-c1e55cd29f1a")
+//let xx = ask.FaceDetectUrl("http://upload-file.haizitong.com/api/getFile/119c44a5-193f-4953-b59c-c1e55cd29f1a")
 //let xx = ask.FaceDetectUrl("https://min.haizitong.com/2/ali/i/fbad726e022742a78a2666580a9a0f0f139317")
-//
+//http://upload-file.haizitong.com/api/getCacheFile/62e23c73991e856c4411bbe4
+//let xx = ask.FaceDetectUrl("https://upload-file.haizitong.com/api/getCacheFile/62e23c73991e856c4411bbe4")
+
+const urls = [
+    "http://image.haizitong.com/cc123be3667143e3af5c53a5906c6b42",
+    "http://upload-file.haizitong.com/api/getFile/119c44a5-193f-4953-b59c-c1e55cd29f1a",
+    "https://min.haizitong.com/2/ali/i/fbad726e022742a78a2666580a9a0f0f139317",
+    "http://upload-file.haizitong.com/api/getCacheFile/62e23c73991e856c4411bbe4"
+]
+for(let o of urls) {
+    let xx2 = ask.FaceDetectUrl(o);
+    console.log("init end", xx2, Date.now() - start);
+}
 
-console.log("init end", xx, Date.now() - start);
+//console.log("init end", xx, Date.now() - start);
 
 // for(let i = 0; i < 10000; ++i) {
 //     let num = ask.FaceFeatureExtractUrl("http://min.haizitong.com/2/ali/i/a5f91d52783d49989777e8b82b545e2c");