def check_current_folder(self, current_set):
self.current_set = current_set
print('==> Checking {}'.format(current_set))
print(
magenta(
'1: D5 / 2: P20 / 3: ZED_rgb / 4: ZED_depth / 5:ZED_confidence'
))
self.get_images_in_current_set(current_set)
self.show_images('1')
while True:
key = cv2.waitKey()
if self.zed_depth_exist:
if chr(key) not in ['1', '2', '3', '4', '5']:
print(red('Pressed key is {}'.format(chr(key))))
break
else:
if chr(key) not in ['1', '2', '3']:
print(red('Pressed key is {}'.format(chr(key))))
break
cv2.destroyAllWindows()
self.show_images(chr(key))
return key
def check_images(self):
self.print_helpping_messages()
# Check if there is folder need to be inspectedssss
if len(self.set_list) == 0:
print(red('No sets to inspect - please check the folder'))
return -1
set_pointer = self.set_list[-1]
while True:
print(len(self.set_list))
# Check if there is folder need to be inspectedssss
if len(self.set_list) == 0:
print(red('No more sets to inspect '))
break
new_key = self.check_current_folder(set_pointer)
current_set_index = self.set_list.index(self.current_set)
if chr(new_key) == 'n':
if current_set_index < len(self.set_list) - 1:
set_pointer = self.set_list[current_set_index + 1]
else:
print(red('This set is the end of the list'))
elif chr(new_key) == 'p':
if current_set_index > 0:
set_pointer = self.set_list[current_set_index - 1]
else:
print(red('This set is the first of the list'))
elif chr(new_key) in ['s', 'h', 'd']:
set_pointer = self.move_items_accordingly(new_key)
elif chr(new_key) == 'q':
print('Exiting the program')
break
else:
print(red('[!]Please press the right key'))
self.print_helpping_messages()
default=False,
help="When indicated, ZED will run.")
args = parser.parse_args()
__ADJUST_TIME__ = args.zed_rest
processes = []
if args.zed:
processes.append('zed')
if args.p20:
processes.append('p20')
if args.d5:
processes.append('D5')
if len(processes) == 0:
print(
red('[Error] Select at least one device to run: --p20 / --zed / --d5'
))
exit(-1)
print(cyan('============== Selected Devices ================'))
print(cyan('{}'.format(processes)))
print(cyan('==================== Guide ====================='))
print(cyan('Mouse LEFT click to record videos.'))
print(cyan('Mouse RIGHT click to exit the program.'))
make_dir_if_not_exists(__ZED_IMG_PATH__)
make_dir_if_not_exists(__P20_IMG_PATH__)
make_dir_if_not_exists(__D5_IMG_PATH__)
# Create logger
img_logger = logutils.create_logger(__VID_LOGGING_FILE__)
"--mode",
type=str,
help="Which dataset that you want to sort out. [IMG | VID]")
parser.add_argument('--set_continue',
default=0,
type=int,
help='Continue to create the set from this number.')
parser.add_argument('--convertSVO', action='store_true')
args = parser.parse_args()
date = args.date
mode = args.mode
set_count = args.set_continue
if mode != 'IMG' and mode != 'VID':
print(red('Please choose the mode between [ IMG | VID ]'))
exit(-1)
if mode == 'VID':
__INPUT_DIR__ = './saved_data/VID/{}/'.format(date)
__LOG_FILE__ = './saved_data/VID/{}/video_capture.log'.format(date)
__OUTPUT_DIR__ = './saved_data/VID/{}/sorted'.format(date)
#
# # list is in [D5, P20, ZED] order
# list_vid_dir = logutils.read_log_file(__LOG_FILE__, __INPUT_DIR__, is_video=True)
#
# put_videos_pairs_in_folder(list_vid_dir, __OUTPUT_DIR__, set_count)
#
# if args.convertSVO:
# make_svo2avi(__OUTPUT_DIR__)
import os
import cv2
from utils.print_utils import yellow, red
__BASE_DIR__ = '/media/dc2019/My Book/VID/VQM_data'
__TEST_DIR__ = os.path.join(__BASE_DIR__, 'zed/test')
__VALID_DIR__ = os.path.join(__BASE_DIR__, 'zed/validation')
sets = os.listdir(__TEST_DIR__)
for set in sets:
set_dir = os.path.join(__TEST_DIR__, set)
print(yellow(set_dir))
frames = os.listdir(set_dir)
for frame in frames:
print(os.path.join(set_dir, frame))
try:
img = cv2.imread(os.path.join(set_dir, frame))
except:
print(red(os.path.join(set_dir, frame)))
print('Done!')
parser.add_argument("--d5", action="store_true", default=False, help="When indicated, Canon D5 Mark IV will run.")
parser.add_argument("--zed", action="store_true", default=False, help="When indicated, ZED will run.")
args = parser.parse_args()
__VID_TIME__ = args.vid_time
__ADJUST_TIME__ = args.zed_rest
processes = []
if args.zed:
processes.append('zed')
if args.p20:
processes.append('p20')
if args.d5:
processes.append('D5')
if len(processes) == 0:
print(red('Select at least one device to run: --p20 / --zed / --d5'))
exit(-1)
print(cyan('============== Selected Devices ================'))
print(cyan('{}'.format(processes)))
print(cyan('==================== Guide ====================='))
print(cyan('Mouse LEFT click to record videos.'))
print(cyan('Mouse RIGHT click to exit the program.'))
make_dir_if_not_exists(__ZED_VID_PATH__)
make_dir_if_not_exists(__P20_VID_PATH__)
make_dir_if_not_exists(__D5_VID_PATH__)
# Create logger
vid_logger = logutils.create_logger(__VID_LOGGING_FILE__)