Пример #1
0
 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()
Пример #2
0
    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 = {}
Пример #3
0
# 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', \