add python layers

voidrank · Dec 17, 2016 · ff3aba6 · ff3aba6
1 parent 0274151
commit ff3aba6
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/python_layers/data/__init__.py b/python_layers/data/__init__.py
diff --git a/python_layers/data/layers.py b/python_layers/data/layers.py
@@ -0,0 +1,16 @@
+import caffe
+import os
+
+import config
+import numpy as np
+from spiders.coco_ssm_spider import * 
+from alchemy.engines.caffe_python_layers import AlchemyDataLayer
+
+
+class COCOSSMDataLayer(AlchemyDataLayer):
+
+    spider = COCOSSMSpider
+    max_cache_item_num = 500
+    wait_time = 10
+    process_num = config.DATA_PROVIDER_PROCESS
+