Files update

simaiden · Dec 28, 2019 · f4a2c1e · f4a2c1e
1 parent 9e6c355
commit f4a2c1e
Show file tree

Hide file tree

Showing 7 changed files with 10 additions and 139 deletions.
diff --git a/.gitignore b/.gitignore
@@ -29,4 +29,6 @@ yolo_df2_shop_descriptors_retrieval_test_4.npy
 yolo_df2_shop_descriptors.npy
 asdasd.py
 retrieval
-Deepfashion2Val
+Deepfashion2Val
+*.eps
+grafico_clases.py
diff --git a/image_yolo_demo.py b/image_yolo_demo.py
diff --git a/log_map.txt b/log_map.txt
diff --git a/new_image_demo.py b/new_image_demo.py
@@ -55,13 +55,13 @@
 
 
 #Faster RCNN / RetinaNet / Mask RCNN
-#model = 'faster'
-#detectron = Predictor(model=model,dataset= dataset, CATEGORIES = classes)
+model = 'maskrcnn'
+detectron = Predictor(model=model,dataset= dataset, CATEGORIES = classes)
 
 #YOLO
 
-detectron = YOLOv3Predictor(params=yolo_params)
-model = 'yolo'
+#detectron = YOLOv3Predictor(params=yolo_params)
+#model = 'yolo'
 
 
 while(True):

diff --git a/resultados_por_modelo_deepfashion2.pkl b/resultados_por_modelo_deepfashion2.pkl
diff --git a/test_times.py b/test_times.py
@@ -17,6 +17,8 @@
 plt.xticks(list(range(1,6)),['YOLOv3','RetinaNet','Faster R-CNN','Mask R-CNN','TridentNet'])
 for i, v in enumerate(mean):
     plt.text(list(range(1,6))[i] - 0.25, v + 0.005, str(round(v,3)))
-plt.ylabel('segundos')
+plt.ylabel('Segundos')
+plt.xlabel('Modelo')
 plt.title('Tiempos promedio de inferencia modelos entrenados')
 plt.show()
+plt.savefig('tiempos.eps')
diff --git a/tests/amazon.jpg b/tests/amazon.jpg