Merge pull request #29 from gumgum/develop

Develop

multivitamin/applications/images/detectors/mm_detector.py

@@ -88,6 +88,11 @@ def __init__(
             raise ValueError("config_file and/or model_file does not exist")
         try:
             self.model = init_detector(config_file, model_file, device="cuda:{}".format(gpuid))
+            dummy_image= np.zeros((720, 1280, 3), dtype = "uint8")
+            images=[dummy_image]
+            tstamps=[0.0]
+            self.process_images(images, tstamps)
+            log.info('dummy forward pass successful')
         except:
             raise ValueError("Could not init_detector")
 

multivitamin/version.py

@@ -1 +1 @@
-__version__ = "1.3.35"
+__version__ = "1.3.36"

tools/push_sqs.py

@@ -0,0 +1,21 @@
+import sys
+import argparse
+
+import boto3
+
+sqs = boto3.resource('sqs')
+a = argparse.ArgumentParser()
+a.add_argument("--queue_name")
+a.add_argument("--request_txt")
+args = a.parse_args()
+
+reqs = []
+with open(args.request_txt) as rf:
+    reqs = rf.readlines()
+
+reqs = [r.strip() for r in reqs]
+queue = sqs.get_queue_by_name(QueueName=args.queue_name)
+for req in reqs:
+    resp = queue.send_message(MessageBody=req)
+    print(resp)
+