def bench(fn, df, writer, reader, desc=''):
twrite = Timer(start=1)
res = writer(df, fn)
if res is INFEASIBLE:
# print('INFEASIBLE')
return nan, nan, nan
twrite.end()
# print('Written with', writer)
tread = Timer(start=1)
rdf = reader(fn)
tread.end()
assert df.shape == rdf.shape, '{} != {}'.format(df.shape, rdf.shape)
# assert (df.dtypes == rdf.dtypes).all(), '{}\n\n != \n\n{}'.format(df.dtypes, rdf.dtypes)
return twrite.time, tread.time, getsize(fn) / 10**6
def main():
if len(sys.argv) == 1:
line()
print(
"To see more details about the available arguments, enter 'python main.py -h'"
)
line()
parser = ArgumentParser(formatter_class=RawTextHelpFormatter)
# Original video path.
parser.add_argument(
'-ovp',
'--original-video-path',
type=str,
required=True,
help='Enter the path of the original '
'video. A relative or absolute path can be specified. '
'If the path contains a space, it must be surrounded in double quotes.\n'
'Example: -ovp "C:/Users/H/Desktop/file 1.mp4"')
# Set the number of threads to be used when computing VMAF.
parser.add_argument(
'--threads',
type=str,
default=str(os.cpu_count()),
help='Set the number of threads to be used when computing VMAF.')
# Which video encoder to use.
parser.add_argument(
'-e',
'--video-encoder',
type=str,
default='x264',
choices=['x264', 'x265', 'av1'],
help='Specify the encoder to use (default: x264).\nExample: -e x265')
# Set AV1 speed/quality ratio
parser.add_argument(
'--cpu-used',
type=int,
default=1,
choices=range(1, 9),
help=
'Only applicable if choosing the AV1 encoder. Set speed/quality ratio. Value Range: 1-8\n'
'Lower values mean slower encoding but better quality, and vice-versa.'
)
# CRF value(s).
parser.add_argument('-crf',
'--crf-value',
type=int,
nargs='+',
choices=range(0, 52),
default=23,
help='Specify the CRF value(s) to use.',
metavar='CRF_VALUE(s)')
# Preset(s).
parser.add_argument('-p',
'--preset',
type=str,
nargs='+',
choices=[
'veryslow', 'slower', 'slow', 'medium', 'fast',
'faster', 'veryfast', 'superfast', 'ultrafast'
],
default='medium',
help='Specify the preset(s) to use.',
metavar='PRESET(s)')
# The time interval to use when creating the overview video.
parser.add_argument(
'-i',
'--interval',
type=int,
choices=range(1, 600),
default=0,
help=
'Create a lossless overview video by grabbing a <cliplength> seconds long segment '
'every <interval> seconds from the original video and use this overview video '
'as the "original" video that the transcodes are compared with.\nExample: -i 30',
metavar='<an integer between 1 and 600>')
# The length of each clip.
parser.add_argument(
'-cl',
'--clip-length',
type=int,
choices=range(1, 60),
default=1,
help=
'Defines the length of the clips. Only applies when used with -i > 0. Default: 1.\n'
'Example: -cl 10',
metavar='<an integer between 1 and 60>')
# Use only the first x seconds of the original video.
parser.add_argument(
'-t',
'--encode-length',
type=str,
help=
'Create a lossless version of the original video that is just the first x seconds of the '
'video, use the cut version as the reference and for all encodes. '
'You cannot use this option in conjunction with the -i or -cl arguments.'
'Example: -t 60')
# Enable phone model?
parser.add_argument('-pm',
'--phone-model',
action='store_true',
help='Enable VMAF phone model.')
# Number of decimal places to use for the data.
parser.add_argument(
'-dp',
'--decimal-places',
type=int,
default=2,
help='The number of decimal places to use for the data '
'in the table (default: 2).\nExample: -dp 3')
# Calculate SSIM?
parser.add_argument('-ssim',
'--calculate-ssim',
action='store_true',
help='Calculate SSIM in addition to VMAF.')
# Calculate PSNR?
parser.add_argument('-psnr',
'--calculate-psnr',
action='store_true',
help='Calculate PSNR in addition to VMAF.')
# No transcoding mode.
parser.add_argument(
'-ntm',
'--no-transcoding-mode',
action='store_true',
help=
'Use this mode if you\'ve already transcoded a video and would like its VMAF and '
'(optionally) the SSIM and PSNR to be calculated.\n'
'Example: -ntm -tvp transcoded.mp4 -ovp original.mp4 -ssim -psnr')
# Transcoded video path (only applicable when using the -ntm mode).
parser.add_argument(
'-tvp',
'--transcoded-video-path',
help=
'The path of the transcoded video (only applicable when using the -ntm mode).'
)
args = parser.parse_args()
if args.calculate_psnr:
exit_program(
'PSNR calculation is currently unavailable due to a change that was made in libvmaf v2.0.0.\n'
'Visit https://github.com/Netflix/vmaf/issues/787 for more information.\n'
'You can re-run your command without the psnr argument, but PSNR values will not be calculated.'
)
args_validator = ArgumentsValidator()
validation_result, validation_errors = args_validator.validate(args)
if not validation_result:
for error in validation_errors:
print(f'Error: {error}')
exit_program('Argument validation failed.')
decimal_places = args.decimal_places
original_video_path = args.original_video_path
filename = Path(original_video_path).name
output_folder = f'({filename})'
os.makedirs(output_folder, exist_ok=True)
# this includes the dot eg '.mp4'
output_ext = Path(original_video_path).suffix
# The M4V container does not support the H.265 codec.
if output_ext == '.m4v' and args.video_encoder == 'x265':
output_ext = '.mp4'
# Use class VideoInfoProvider to get the framerate, bitrate and duration
provider = VideoInfoProvider(original_video_path)
fps = provider.get_framerate_fraction()
fps_float = provider.get_framerate_float()
original_bitrate = provider.get_bitrate()
line()
print(
'Video Quality Metrics\nGitHub.com/BassThatHertz/video-quality-metrics'
)
line()
print('Here\'s some information about the original video:')
print(f'Filename: {filename}')
print(f'Bitrate: {original_bitrate}')
print(f'Framerate: {fps} ({fps_float}) FPS')
line()
table = PrettyTable()
table_column_names = ['Encoding Time (s)', 'Size', 'Bitrate', 'VMAF']
if args.calculate_ssim:
table_column_names.append('SSIM')
if args.calculate_psnr:
table_column_names.append('PSNR')
if args.no_transcoding_mode:
del table_column_names[0]
if args.interval > 0:
clip_length = str(args.clip_length)
result, concatenated_video = create_movie_overview(
original_video_path, output_folder, args.interval, clip_length)
if result:
original_video_path = concatenated_video
else:
exit_program(
'Something went wrong when trying to create the overview video.'
)
factory = FfmpegProcessFactory()
timer = Timer()
if not args.no_transcoding_mode:
# args.crf_value is a list when more than one CRF value is specified.
if is_list(args.crf_value) and len(args.crf_value) > 1:
print('CRF comparison mode activated.')
crf_values = args.crf_value
crf_values_string = ', '.join(str(crf) for crf in crf_values)
preset = args.preset[0] if is_list(args.preset) else args.preset
print(
f'CRF values {crf_values_string} will be compared and the {preset} preset will be used.'
)
video_encoder = args.video_encoder
# Cannot use os.path.join for output_folder as this gives an error like the following:
# No such file or directory: '(2.mkv)\\Presets comparison at CRF 23/Raw JSON Data/superfast.json'
output_folder = f'({filename})/CRF comparison at preset {preset}'
os.makedirs(output_folder, exist_ok=True)
# The comparison table will be in the following path:
comparison_table = os.path.join(output_folder, 'Table.txt')
# Add a CRF column.
table_column_names.insert(0, 'CRF')
# Set the names of the columns
table.field_names = table_column_names
# The user only wants to transcode the first x seconds of the video.
if args.encode_length and args.interval == 0:
original_video_path = cut_video(filename, args, output_ext,
output_folder,
comparison_table)
# Transcode the video with each CRF value.
for crf in crf_values:
transcode_output_path = os.path.join(output_folder,
f'CRF {crf}{output_ext}')
graph_filename = f'CRF {crf} at preset {preset}'
arguments = EncodingArguments()
arguments.infile = str(original_video_path)
arguments.encoder = Encoder[video_encoder]
if video_encoder == 'av1':
arguments.av1_compression = str(args.cpu_used)
arguments.crf = str(crf)
arguments.preset = preset
arguments.outfile = transcode_output_path
process = factory.create_process(arguments)
print(f'Transcoding the video with CRF {crf}...')
timer.start()
process.run()
time_rounded = timer.end(decimal_places)
print('Done!')
transcode_size = os.path.getsize(
transcode_output_path) / 1_000_000
transcoded_bitrate = provider.get_bitrate(
transcode_output_path)
size_rounded = force_decimal_places(
round(transcode_size, decimal_places), decimal_places)
data_for_current_row = [
f'{size_rounded} MB', transcoded_bitrate
]
os.makedirs(os.path.join(output_folder, 'Raw JSON Data'),
exist_ok=True)
# os.path.join doesn't work with libvmaf's log_path option so we're manually defining the path with
# slashes.
json_file_path = f'{output_folder}/Raw JSON Data/CRF {crf}.json'
run_libvmaf(transcode_output_path, args, json_file_path, fps,
original_video_path, factory)
create_table_plot_metrics(comparison_table, json_file_path,
args, decimal_places,
data_for_current_row, graph_filename,
table, output_folder, time_rounded,
crf)
with open(comparison_table, 'a') as f:
f.write(f'\nFile Transcoded: {filename}')
f.write(f'\nBitrate: {original_bitrate}')
f.write(f'\nEncoder used for the transcodes: {video_encoder}')
f.write(f'\nPreset used for the transcodes: {preset}')
# args.preset is a list when more than one preset is specified.
elif is_list(args.preset):
print('Presets comparison mode activated.')
chosen_presets = args.preset
presets_string = ', '.join(chosen_presets)
crf = args.crf_value[0] if is_list(
args.crf_value) else args.crf_value
video_encoder = args.video_encoder
print(
f'Presets {presets_string} will be compared at a CRF of {crf}.'
)
# Cannot use os.path.join for output_folder as this gives an error like the following:
# No such file or directory: '(2.mkv)\\Presets comparison at CRF 23/Raw JSON Data/superfast.json'
output_folder = f'({filename})/Presets comparison at CRF {crf}'
os.makedirs(output_folder, exist_ok=True)
comparison_table = os.path.join(output_folder, 'Table.txt')
table_column_names.insert(0, 'Preset')
# Set the names of the columns
table.field_names = table_column_names
# The user only wants to transcode the first x seconds of the video.
if args.encode_length:
original_video_path = cut_video(filename, args, output_ext,
output_folder,
comparison_table)
# Transcode the video with each preset.
for preset in chosen_presets:
transcode_output_path = os.path.join(output_folder,
f'{preset}{output_ext}')
graph_filename = f"Preset '{preset}'"
arguments = EncodingArguments()
arguments.infile = original_video_path
arguments.encoder = Encoder[video_encoder]
arguments.crf = str(crf)
arguments.preset = preset
arguments.outfile = transcode_output_path
process = factory.create_process(arguments)
line()
print(f'Transcoding the video with preset {preset}...')
timer.start()
process.run()
time_rounded = timer.end(decimal_places)
print('Done!')
transcode_size = os.path.getsize(
transcode_output_path) / 1_000_000
transcoded_bitrate = provider.get_bitrate(
transcode_output_path)
size_rounded = force_decimal_places(
round(transcode_size, decimal_places), decimal_places)
data_for_current_row = [
f'{size_rounded} MB', transcoded_bitrate
]
os.makedirs(os.path.join(output_folder, 'Raw JSON Data'),
exist_ok=True)
# os.path.join doesn't work with libvmaf's log_path option so we're manually defining the path with
# slashes.
json_file_path = f'{output_folder}/Raw JSON Data/{preset}.json'
run_libvmaf(transcode_output_path, args, json_file_path, fps,
original_video_path, factory)
create_table_plot_metrics(comparison_table, json_file_path,
args, decimal_places,
data_for_current_row, graph_filename,
table, output_folder, time_rounded,
preset)
with open(comparison_table, 'a') as f:
f.write(f'\nFile Transcoded: {filename}')
f.write(f'\nBitrate: {original_bitrate}')
f.write(f'\nEncoder used for the transcodes: {video_encoder}')
f.write(f'\nCRF value used for the transcodes: {crf}')
# -ntm argument was specified.
else:
line()
output_folder = f'({filename})'
os.makedirs(output_folder, exist_ok=True)
comparison_table = os.path.join(output_folder, 'Table.txt')
table.field_names = table_column_names
# os.path.join doesn't work with libvmaf's log_path option so we're manually defining the path with slashes.
json_file_path = f'{output_folder}/QualityMetrics.json'
# Run libvmaf to get the quality metric(s).
run_libvmaf(args.transcoded_video_path, args, json_file_path, fps,
original_video_path, factory)
transcode_size = os.path.getsize(
args.transcoded_video_path) / 1_000_000
size_rounded = force_decimal_places(
round(transcode_size, decimal_places), decimal_places)
transcoded_bitrate = provider.get_bitrate(args.transcoded_video_path)
data_for_current_row = [f'{size_rounded} MB', transcoded_bitrate]
graph_filename = 'The variation of the quality of the transcoded video throughout the video'
# Create the table and plot the metrics if -dqm was not specified.
create_table_plot_metrics(json_file_path,
args,
decimal_places,
data_for_current_row,
graph_filename,
table,
output_folder,
time_rounded=None,
crf_or_preset=None)
line()
print(f'All done! Check out the ({filename}) folder.')
from utils import Timer
SIZE = 3
kernel = np.ones((SIZE, SIZE)) / (SIZE * SIZE)
# = [[1/9, 1/9, 1/9],
# [1/9, 1/9, 1/9], (SIZE = 3)
# [1/9, 1/9, 1/9]]
cap = cv2.VideoCapture(0)
if not cap.isOpened():
sys.exit("No cameras available!")
timer = Timer()
while True:
ret, img_orig = cap.read()
img_src = cv2.cvtColor(img_orig, cv2.COLOR_BGR2GRAY)
timer.start()
img_dst = cv2.filter2D(img_src, -1, kernel)
timer.end(img_dst)
cv2.imshow("Before", img_src)
cv2.imshow("After", img_dst)
if cv2.waitKey(10) & 0xFF == ord("q"):
break
cv2.destroyAllWindows()
from bot import ShrimpBot
from utils import Timer
from const import Settings
from logger import get_logger
timer = Timer()
logger = get_logger("shrimp_bot")
ShrimpBot(admin=Settings.Admins, logger=logger).run(Settings.token)
logger.info("Run time - %dh : %dm : %ds" % timer.end())