Atualizando README

README.md

@@ -19,6 +19,7 @@ Pacote de Visão Computacional do Inovisão.
   ```
 
 - Uma imagem como a mostrada abaixo deve ser apresentada:
+
     ![pynovisao](data/pynovisao.png)
     
 ## Outras Opções
@@ -26,17 +27,17 @@ Pacote de Visão Computacional do Inovisão.
 - Mostra todas as opções disponíveis
 
   ```
-  $ python slicParametros.py --help
+  $ python main.py --help
   ```
 - Executa o programa inicializando o banco de imagens em *../data/soja*
 
   ```
-  $ python slicParametros.py --dataset ../data/digits
+  $ python main.py --dataset ../data/soja
   ```
 - Executa o programa definindo as classes e suas respectivas cores (X11 color names)
 
   ```
-  $ python slicParametros.py --classes "Solo Soja Gramineas FolhasLargas" --colors "Orange SpringGreen RebeccaPurple Snow"
+  $ python main.py --classes "Solo Soja Gramineas FolhasLargas" --colors "Orange SpringGreen RebeccaPurple Snow"
   ```
 
 

src/Pynovisao.py

@@ -37,9 +37,15 @@ class Act(object):
         self.tk = tk
         self.segmenter = Slic.Slic()
 
-        self.dataset = args["dataset"]
+        self.__init_dataset(args["dataset"])
         self.__init_classes(args["classes"], args["colors"])
 
+    def __init_dataset(self, directory):
+        if(directory[-1] == '/'):
+            directory = directory[:-1]
+            
+        self.dataset = directory
+        f.create_dir(self.dataset)
     
     def __init_classes(self, classes = None, colors = None):
         self.classes = []
@@ -159,7 +165,7 @@ class Act(object):
     def set_dataset_path(self):
         directory = self.tk.utils.ask_directory(default_dir = self.dataset)
         if directory:
-            self.dataset = directory
+            self.__init_dataset(directory)
             self.tk.write_log("Image dataset defined: %s", self.dataset)
             
             self.__init_classes()

src/util/File.py

@@ -59,3 +59,8 @@ class File(object):
     def list_dirs(dataset):
         return [name for name in os.listdir(dataset) 
                     if os.path.isdir(os.path.join(dataset, name))]
+
+    @staticmethod
+    def create_dir(directory):
+        if not os.path.exists(directory):
+            os.makedirs(directory)
\ No newline at end of file