expose aug_matrix

Author: Ruilong Li

README.md

@@ -5,81 +5,25 @@ Pytorch based data streamer. (Capture, Video & Image).
 ## Install
 
 ```
-pip install git+https://github.com/liruilong940607/streamer_pytorch --upgrade
+pip install git+https://github.com/Project-Splinter/streamer_pytorch --upgrade
 ```
 
 ## Usage
 
 ```
 # images
 python demo.py --images <IMAGE_PATH> <IMAGE_PATH> <IMAGE_PATH> --loop --vis
+# image folder
+python demo.py --image_folder <IMAGE_FOLDER_PATH> --loop --vis
 # videos
 python demo.py --videos <VIDEO_PATH> <VIDEO_PATH> <VIDEO_PATH> --vis
 # capture device
 python demo.py --camera --vis
 ```
 
-see also in `demo.py`
-```
-import tqdm
-import argparse
-import torch
-import numpy as np
-import cv2
-import streamer_pytorch as streamer
-
-parser = argparse.ArgumentParser(description='.')
-parser.add_argument(
-    '--camera', action="store_true")
-parser.add_argument(
-    '--images', default="", nargs="*")
-parser.add_argument(
-    '--videos', default="", nargs="*")
-parser.add_argument(
-    '--loop', action="store_true")
-parser.add_argument(
-    '--vis', action="store_true")
-args = parser.parse_args()
-
-def visulization(data):
-    window = data[0].numpy()
-    window = window.transpose(1, 2, 0)
-    window = (window * 0.5 + 0.5) * 255.0
-    window = np.uint8(window)
-    window = cv2.cvtColor(window, cv2.COLOR_BGR2RGB) 
-    window = cv2.resize(window, (0, 0), fx=2, fy=2)
-
-    cv2.imshow('window', window)
-    cv2.waitKey(1)
-
-if args.camera:
-    data_stream = streamer.CaptureStreamer()
-elif len(args.videos) > 0:
-    data_stream = streamer.VideoListStreamer(
-        args.videos * (100 if args.loop else 1))
-elif len(args.images) > 0:
-    data_stream = streamer.ImageListStreamer(
-        args.images * (100 if args.loop else 1))
-
-loader = torch.utils.data.DataLoader(
-    data_stream, 
-    batch_size=1, 
-    num_workers=1, 
-    pin_memory=False,
-)
-
-try:
-    for data in tqdm.tqdm(loader):
-        if args.vis:
-            visulization(data)
-except Exception as e:
-    print (e)
-    del data_stream
-
-```
-
 ## API
 
 - CaptureStreamer(id=0, width=512, height=512, pad=True)
 - VideoListStreamer(files, width=512, height=512, pad=True)
 - ImageListStreamer(files, width=512, height=512, pad=True)
+- aug_matrix(w1, h1, w2, h2, pad=True)

demo.py

@@ -13,7 +13,7 @@
 parser.add_argument(
     '--images', default="", nargs="*", help="paths of image.")
 parser.add_argument(
-    '--image_folder', default="", help="path of image folder.")
+    '--image_folder', default=None, help="path of image folder.")
 parser.add_argument(
     '--videos', default="", nargs="*", help="paths of video.")
 parser.add_argument(

streamer_pytorch/__init__.py

@@ -2,3 +2,4 @@
 from .streamer import CaptureStreamer
 from .streamer import VideoListStreamer
 from .streamer import ImageListStreamer
+from .streamer import aug_matrix