Image dimensions

darknet.cc

@@ -72,7 +72,7 @@ class ImageDetectionWorker : public AsyncWorker {
   // so it is safe to use V8 again
   void HandleOKCallback () {
     HandleScope scope;
-    const int argc = 3;
+    const int argc = 4;
     v8::Local<v8::Array> results = Nan::New<v8::Array>();
     for(int i = 0; i < output->numberOfResults; i++) {
         RecognitionResult result = output->recognitionResults[i];
@@ -87,10 +87,15 @@ class ImageDetectionWorker : public AsyncWorker {
         i++;
     }
 
+    v8::Local<v8::Object> dimensions = Nan::New<v8::Object>();
+    dimensions->Set(Nan::New("width").ToLocalChecked(), Nan::New<v8::Number>(output->frameWidth));
+    dimensions->Set(Nan::New("height").ToLocalChecked(), Nan::New<v8::Number>(output->frameHeight));
+
     v8::Local<v8::Value> argv[argc] = {
         Nan::NewBuffer(output->modifiedFrame, output->modifiedFrameSize).ToLocalChecked(),
         Nan::NewBuffer(output->originalFrame, output->originalFrameSize).ToLocalChecked(),
         results,
+        dimensions
     };
 
     callback->Call(argc, argv);

src/demo.cc

@@ -340,6 +340,9 @@ WorkerData* start_image_demo(InputOptions opts) {
     progressState->numberOfResults = current_results.size();
     progressState->recognitionResults = new RecognitionResult[progressState->numberOfResults];
 
+    progressState->frameWidth = original.w;
+    progressState->frameHeight = original.h;
+
     int i = 0;
     for(auto const& result: current_results) {
         progressState->recognitionResults[i] = result;

test/darknetImageTest.js

@@ -6,8 +6,8 @@ darknet.detectImage({
   weights: './yolo.weights',
   data: './cfg/coco.data',
   image: './data/dog.jpg',
-}, function(modified, original, detections) {
-  console.log(modified.length, original.length, detections);
+}, function(modified, original, detections, dimensions) {
+  console.log(modified.length, original.length, detections, dimensions);
   fs.writeFileSync('./data.modified.raw', modified);
   // ffmpeg -f rawvideo -s 768x576 -pix_fmt bgr24 -i data.raw data.jpg
   fs.writeFileSync('./data.raw', original);