コード例 #1
0
ファイル: process.py プロジェクト: super3/PyTrack
    def process(self):
        """Process frames in queue/memory."""
        # Start benchmark
        self.bench.start("Processing", len(self.files))
        # Temporary Data Var
        data = []

        # Initialize Compare Object
        if METHOD == algorithm.BACKGROUND_SUBTRACTION:
            # Load Current Frame and Background
            compare = CompareImages(self.queue[0], background)
        elif METHOD == algorithm.FRAME_DIFFERENCING:
            # Load Current Frame and Next Frame
            compare = CompareImages(self.queue[0], self.queue[1])

        # Process Image Queue
        for frame in self.queue:
            if METHOD == algorithm.BACKGROUND_SUBTRACTION:
                # Load Current Frame
                compare.setLeft(frame)
            elif METHOD == algorithm.FRAME_DIFFERENCING:
                # Load Next Frame
                compare.setRight(frame)
            data.append(compare.process(TOLERANCE))

        # End benchmark
        self.bench.end()
        # Write data to file
        listToFile(data, 'data.txt')
コード例 #2
0
ファイル: process.py プロジェクト: super3/PyTrack
	def process(self):
		"""Process frames in queue/memory."""
		# Start benchmark
		self.bench.start("Processing", len(self.files))
		# Temporary Data Var
		data = []

		# Initialize Compare Object
		if METHOD == algorithm.BACKGROUND_SUBTRACTION:
			# Load Current Frame and Background
			compare = CompareImages( self.queue[0], background )
		elif METHOD == algorithm.FRAME_DIFFERENCING:
			# Load Current Frame and Next Frame
			compare = CompareImages( self.queue[0], self.queue[1] )

		# Process Image Queue
		for frame in self.queue:
			if METHOD == algorithm.BACKGROUND_SUBTRACTION:
				# Load Current Frame
				compare.setLeft( frame )
			elif METHOD == algorithm.FRAME_DIFFERENCING:
				# Load Next Frame
				compare.setRight( frame )
			data.append( compare.process( TOLERANCE ) )

		# End benchmark
		self.bench.end()
		# Write data to file
		listToFile(data, 'data.txt')
コード例 #3
0
ファイル: viewer.py プロジェクト: super3/PyTrack
		else:
			tmpImage = ImageFile(files[currentFile]).imgFile

	# If Current File is not the Processed File or a 
	if currentFile != processedFile:
		

		# Get the Images Needed Based on the Algorithm
		try:
			if METHOD == algorithm.BACKGROUND_SUBTRACTION:
				# Load Current Frame
				compare.setLeft( ImageFile(files[currentFile], currentFile) )

			elif METHOD == algorithm.FRAME_DIFFERENCING:
				# Load Next Frame
				compare.setRight( ImageFile(files[currentFile+1], currentFile+1) )

		except IOError as e:
			# Display Screen Error if Images are Missing
			font = pygame.font.SysFont("Consolas", 18)
			tmpImage = font.render("Could not load one or more images.", 1, RED)
		except ValueError as e:
			# Display Screen Error if Images are not Equal Dimensions
			font = pygame.font.SysFont("Consolas", 18)
			tmpImage = font.render("Images are not the same dimensions.", 1, RED)

		else:
			# Do Compare Operation
			compare.process(TOLERANCE)

			# If showSource is enabled then process the frames, and get the diffed result for display
コード例 #4
0
        # Else then process the frames, but get the first frame for display
        else:
            tmpImage = ImageFile(files[currentFile]).imgFile

    # If Current File is not the Processed File or a
    if currentFile != processedFile:

        # Get the Images Needed Based on the Algorithm
        try:
            if METHOD == algorithm.BACKGROUND_SUBTRACTION:
                # Load Current Frame
                compare.setLeft(ImageFile(files[currentFile], currentFile))

            elif METHOD == algorithm.FRAME_DIFFERENCING:
                # Load Next Frame
                compare.setRight(
                    ImageFile(files[currentFile + 1], currentFile + 1))

        except IOError as e:
            # Display Screen Error if Images are Missing
            font = pygame.font.SysFont("Consolas", 18)
            tmpImage = font.render("Could not load one or more images.", 1,
                                   RED)
        except ValueError as e:
            # Display Screen Error if Images are not Equal Dimensions
            font = pygame.font.SysFont("Consolas", 18)
            tmpImage = font.render("Images are not the same dimensions.", 1,
                                   RED)

        else:
            # Do Compare Operation
            compare.process(TOLERANCE)