added jpg save to export annotation images

Author: chriamue

cmake/CompileOptions.cmake

@@ -49,7 +49,7 @@ set(DEFAULT_INCLUDE_DIRECTORIES)
 # Libraries
 # 
 
-set(DEFAULT_LIBRARIES ${Boost_LIBRARIES} ${OpenCV_LIBS} ${Poco_LIBRARIES})
+set(DEFAULT_LIBRARIES stdc++fs ${Boost_LIBRARIES} ${OpenCV_LIBS} ${Poco_LIBRARIES})
 
 endif( OPTION_CONAN_PACKAGES )
 # 

source/annotatorlib/include/AnnotatorLib/Export/AbstractExport.h

@@ -29,6 +29,8 @@ class ANNOTATORLIB_API AbstractExport {
 
   virtual bool equals(std::shared_ptr<AbstractExport> other) = 0;
 
+  virtual void doExport() = 0;
+
   virtual ~AbstractExport() {}
 };
 /************************************************************/

source/annotatorlib/include/AnnotatorLib/Export/ExportAnnotationImages.h

@@ -8,6 +8,7 @@
 
 #include <AnnotatorLib/AnnotatorLibDatastructs.h>
 #include <AnnotatorLib/Export/AbstractExport.h>
+#include <AnnotatorLib/Project.h>
 #include <AnnotatorLib/annotatorlib_api.h>
 
 #include <string>
@@ -16,14 +17,18 @@
 #include <boost/filesystem.hpp>
 
 namespace AnnotatorLib {
+
+class Class;
+class Object;
+
 namespace Export {
 /************************************************************/
 /**
  *
  */
 class ANNOTATORLIB_API ExportAnnotationImages : public AbstractExport {
  public:
-  ExportAnnotationImages(std::string path);
+  ExportAnnotationImages(std::shared_ptr<Project> project, std::string path);
 
   /**
    *
@@ -33,15 +38,16 @@ class ANNOTATORLIB_API ExportAnnotationImages : public AbstractExport {
 
   virtual std::string getPath() override;
 
+  virtual void doExport() override;
+
   virtual bool equals(std::shared_ptr<AbstractExport> other) override;
 
  protected:
+  void exportClass(std::shared_ptr<Class> theClass);
+  void exportObject(std::shared_ptr<Object> object);
 
+  std::shared_ptr<Project> project;
   std::string path;
-
-  long position = 0;
-  std::vector<boost::filesystem::path> images;
-  std::vector<boost::filesystem::path>::const_iterator imgIter;
 };
 /************************************************************/
 /* External declarations (package visibility)               */

source/annotatorlib/include/AnnotatorLib/Export/ExportFactory.h

@@ -13,6 +13,7 @@
 #include <AnnotatorLib/AnnotatorLibDatastructs.h>
 #include <AnnotatorLib/Export/AbstractExport.h>
 #include <AnnotatorLib/Export/ExportPlugin.h>
+#include <AnnotatorLib/Project.h>
 #include <AnnotatorLib/annotatorlib_api.h>
 
 namespace AnnotatorLib {
@@ -30,9 +31,10 @@ class ANNOTATORLIB_API ExportFactory {
    * @return imageSet
    */
   std::shared_ptr<AbstractExport> createExport(std::string type,
-                                                   std::string path);
+                                               std::shared_ptr<Project> project,
+                                               std::string path);
 
-  std::list<std::string> availableImageSets();
+  std::list<std::string> availableExports();
 
   /**
    * @brief loadPlugins

source/annotatorlib/include/AnnotatorLib/Export/ExportPlugin.h

@@ -5,6 +5,7 @@
 #include <string>
 
 #include <AnnotatorLib/Export/AbstractExport.h>
+#include <AnnotatorLib/Project.h>
 #include <AnnotatorLib/annotatorlib_api.h>
 
 namespace AnnotatorLib {
@@ -36,7 +37,7 @@ class ANNOTATORLIB_API ExportPlugin {
    * @return new instance of loader
    */
   virtual std::shared_ptr<AnnotatorLib::Export::AbstractExport> create(
-      std::string path) = 0;
+      std::shared_ptr<Project> project, std::string path) = 0;
 };
 }
 }

source/annotatorlib/source/Export/ExportAnnotationImages.cpp

@@ -3,11 +3,14 @@
 #define Annotator_AnnotatorLib_ExportAnnotationImages_BODY
 
 /************************************************************
- ImageFolder class body
+ ExportAnnotationImages class body
  ************************************************************/
 
 // include associated header file
 
+#include <experimental/filesystem>
+
+#include "AnnotatorLib/Annotation.h"
 #include "AnnotatorLib/Export/ExportAnnotationImages.h"
 
 #include <set>
@@ -17,10 +20,14 @@
 
 // Derived includes directives
 
+namespace fs = std::experimental::filesystem;
+
 namespace AnnotatorLib {
 namespace Export {
 
-ExportAnnotationImages::ExportAnnotationImages(std::string path) {
+ExportAnnotationImages::ExportAnnotationImages(std::shared_ptr<Project> project,
+                                               std::string path) {
+  this->project = project;
   this->path = path;
 }
 
@@ -33,8 +40,42 @@ bool ExportAnnotationImages::equals(std::shared_ptr<AbstractExport> other) {
   return true;
 }
 
+void ExportAnnotationImages::exportClass(std::shared_ptr<Class> theClass) {
+  std::string classPath = path + "/" + theClass->getName();
+}
+
+void ExportAnnotationImages::exportObject(std::shared_ptr<Object> object) {
+  std::string classPath = path + "/" + object->getClass()->getName() + "/";
+  if (!fs::exists(classPath)) {
+    fs::create_directory(classPath);
+  }
+  std::string objectPath = classPath + object->getName() + "_";
+  for (auto annotation : object->getAnnotations()) {
+    unsigned long frame =
+        annotation.second.lock()->getFrame()->getFrameNumber();
+    cv::Mat tmp = project->getImageSet()->getImage(frame);
+    float x = std::max(annotation.second.lock()->getX(), 0.f);
+    float y = std::max(annotation.second.lock()->getY(), 0.f);
+    float w = std::min(annotation.second.lock()->getWidth(), tmp.cols - x);
+    float h = std::min(annotation.second.lock()->getHeight(), tmp.rows - y);
+
+    cv::Rect rect(x, y, w, h);
+
+    cv::Mat cropped;
+    tmp(rect).copyTo(cropped);
+    std::string imagePath = objectPath + std::to_string(frame) + ".jpg";
+    cv::imwrite(imagePath, cropped);
+  }
+}
+
 std::string ExportAnnotationImages::getPath() { return path; }
 
+void ExportAnnotationImages::doExport() {
+  for (auto object : project->getSession()->getObjects()) {
+    exportObject(object.second);
+  }
+}
+
 // static attributes (if any)
 }  // of namespace Export
 }  // of namespace AnnotatorLib

source/annotatorlib/source/Export/ExportFactory.cpp

@@ -33,16 +33,16 @@ static ExportFactory *_instance = nullptr;
  * @return imageSet
  */
 std::shared_ptr<AbstractExport> ExportFactory::createExport(
-    std::string type, std::string path) {
+    std::string type, std::shared_ptr<Project> project, std::string path) {
   if ("annotationimages" == type)
-    return std::make_shared<ExportAnnotationImages>(path);
+    return std::make_shared<ExportAnnotationImages>(project, path);
 
   std::shared_ptr<ExportPlugin> plugin = this->plugins[type];
-  if (plugin) return plugin->create(path);
+  if (plugin) return plugin->create(project, path);
   throw std::invalid_argument("ExportType unknown!");
 }
 
-std::list<std::string> ExportFactory::availableImageSets() {
+std::list<std::string> ExportFactory::availableExports() {
   return _availableExports;
 }
 
@@ -90,9 +90,8 @@ ExportFactory::~ExportFactory() { _instance = nullptr; }
 
 void ExportFactory::addAvailableExport(ExportPlugin *plugin) {
   if (plugin) {
-    bool found =
-        (std::find(_availableExports.begin(), _availableExports.end(),
-                   plugin->name()) != _availableExports.end());
+    bool found = (std::find(_availableExports.begin(), _availableExports.end(),
+                            plugin->name()) != _availableExports.end());
     if (!found)
       _availableExports.insert(_availableExports.end(), plugin->name());
   }