def __init__(self, process): self._process = process self.running = False self.stopped = True self._count = 0 self.count = self._count self._meta = [] self.meta = self._meta.copy() self.event = Event() self.fps = Rate()
def __init__(self, process): self._process = process self.running = False self.stopped = True self._count = 0 self.count = self._count self._srcid = 0 self.srcid = self._srcid self._meta = [] self.meta = self._meta.copy() self._img = [] self.outimg = self._img.copy() self._timestamp = 0 self.timestamp = self._timestamp self.event = Event() self.fps = Rate() self.history = {}
# Parse the args args = vars(parser.parse_args()) context = zmq.Context() footage_socket = context.socket(zmq.SUB) footage_socket.bind('tcp://' + args['address'] + ':' + args['port']) footage_socket.setsockopt_string(zmq.SUBSCRIBE, '') footage_socket.set_hwm(1) footage_socket.RCVTIMEO = 1000 # in milliseconds count = 0 running = True from rocketvision import Rate fps = Rate() fps.start() from nada import Nada from yolo import Yolo from resnet50 import ResNet50 nada = Nada() #nn = ResNet50() nn = Yolo( img_size=256, conf_thres=0.60 ) # default is 512 which yeilds about 3.8 fps (i7/940MX), 384 --> 5 fps, 256 --> 7 fps # nn2 = Yolo(img_size=256, conf_thres = 0.60) # default is 512 which yeilds about 3.8 fps (i7/940MX), 384 --> 5 fps, 256 --> 7 fps # nn = Yolo(cfg='ultrayolo/cfg/yolov3-tiny.cfg', \