__author__ = "Simon Nilsson; sronilsson@gmail.com"
import os
from subprocess import PIPE, Popen
from tkinter import *
from tkinter import filedialog
from typing import Optional, Union
import matplotlib
matplotlib.use('agg')
import cv2
import pandas as pd
from PIL import Image, ImageTk
from tabulate import tabulate
import simba
from simba.mixins.config_reader import ConfigReader
from simba.utils.checks import check_file_exist_and_readable, check_int
from simba.utils.enums import Formats, Options, TagNames
from simba.utils.errors import (AdvancedLabellingError, FrameRangeError,
NoDataError)
from simba.utils.lookups import (get_labelling_img_kbd_bindings,
get_labelling_video_kbd_bindings)
from simba.utils.printing import log_event, stdout_success
from simba.utils.read_write import (get_all_clf_names, get_fn_ext,
get_video_meta_data, read_config_entry,
read_df)
[docs]class AdvancedLabellingInterface(ConfigReader):
"""
Launch advanced labelling (annotation) interface in SimBA.
.. note::
`Advanced annotation tutorial <https://github.com/sgoldenlab/simba/blob/master/docs/advanced_labelling.md>`__.
.. image:: _static/img/adv_annotator.png
:alt: Adv annotator
:width: 800
:align: center
:param str config_path: path to SimBA project config file in Configparser format
:param str file_path: Path to video that is to be annotated
:param bool continuing: If True, user is continuing the annotations of a video with started but incomplete annotations.
Examples
----------
>>> select_labelling_video_advanced(config_path='MyProjectConfig', file_path='MyVideoFilePath', continuing=True)
"""
def __init__(self, config_path: str, file_path: str, continuing: bool):
ConfigReader.__init__(self, config_path=config_path)
log_event(logger_name=str(self.__class__.__name__), log_type=TagNames.CLASS_INIT.value, msg=self.create_log_msg_from_init_args(locals=locals()))
self.padding, self.file_path = 5, file_path
self.frm_no, self.video_path = 0, file_path
self.continuing = continuing
self.play_video_script_path = os.path.join(os.path.dirname(simba.__file__), "labelling", "play_annotation_video.py")
_, self.video_name, _ = get_fn_ext(filepath=file_path)
self.features_extracted_file_path = os.path.join(self.features_dir, self.video_name + "." + self.file_type)
self.targets_inserted_file_path = os.path.join(self.targets_folder, self.video_name + "." + self.file_type)
self.machine_results_file_path = os.path.join(self.machine_results_dir, self.video_name + "." + self.file_type)
self.cap = cv2.VideoCapture(self.video_path)
self.img_kbd_bindings = get_labelling_img_kbd_bindings()
self.video_kbd_bindings = get_labelling_video_kbd_bindings()
self.__create_key_presses_lbl()
self.__create_video_key_presses_lbl()
self.video_meta_data = get_video_meta_data(video_path=self.video_path)
self.frame_lst = list(range(0, self.video_meta_data["frame_count"]))
self.max_frm_no = max(self.frame_lst)
self.target_lst = get_all_clf_names(config=self.config, target_cnt=self.clf_cnt)
if len(self.target_lst) == 0:
raise NoDataError(msg='To annotate behaviors, your SimBA project needs at least one defined classifier. Found 0', source=self.__class__.__name__)
self.max_frm_size = 1080, 650
self.main_window = Toplevel()
if continuing:
check_file_exist_and_readable(file_path=self.targets_inserted_file_path)
check_file_exist_and_readable(file_path=self.features_extracted_file_path)
if self.file_type == Formats.CSV.value:
self.data_df = (pd.read_csv(self.targets_inserted_file_path).set_index("Unnamed: 0")[self.target_lst].astype(int))
self.data_df.index.name = None
if self.file_type == Formats.PARQUET.value:
self.data_df = pd.read_parquet(self.targets_inserted_file_path)[self.target_lst]
self.data_df_features = read_df(self.features_extracted_file_path, self.file_type)
self.data_df = self.data_df_features.join(self.data_df)
self.main_window.title("SIMBA ANNOTATION INTERFACE (CONTINUING ANNOTATIONS)")
try:
self.frm_no = read_config_entry(self.config, "Last annotated frames", self.video_name.lower(), data_type="int")
except ValueError:
pass
else:
check_file_exist_and_readable(file_path=self.features_extracted_file_path)
self.data_df = read_df(self.features_extracted_file_path, self.file_type)
for target in self.target_lst:
self.data_df[target] = None
self.main_window.title("SIMBA ANNOTATION INTERFACE (ADVANCED ANNOTATION) - VIDEO {}".format(self.video_name))
self.data_df_targets = self.data_df[self.target_lst]
self.folder = Frame(self.main_window)
self.buttons_frm = Frame(self.main_window, bd=2, width=700, height=300)
self.current_frm_n = IntVar(self.main_window, value=self.frm_no)
self.change_frm_box = Entry(self.buttons_frm, width=7, textvariable=self.current_frm_n)
self.frame_number_lbl = Label(self.buttons_frm, text="Frame number")
self.forward_btn = Button(self.buttons_frm, text=">", command=lambda: self.advance_frame(new_frm_number=int(self.current_frm_n.get() + 1)))
self.backward_btn = Button(self.buttons_frm, text="<", command=lambda: self.advance_frame(new_frm_number=int(self.current_frm_n.get() - 1)))
self.forward_max_btn = Button(self.buttons_frm, text=">>", command=lambda: self.advance_frame(len(self.frame_lst) - 1))
self.next_absent_forward_btn = Button(self.buttons_frm, text="NEXT ABSENT", fg="red", command=lambda: self.find_last_next_annotation(forwards=True, present=False))
self.next_present_forward_btn = Button(self.buttons_frm, text="NEXT PRESENT", fg="blue", command=lambda: self.find_last_next_annotation(forwards=True, present=True))
self.last_absent_forward_btn = Button(self.buttons_frm, text="PREVIOUS ABSENT", fg="red", command=lambda: self.find_last_next_annotation(forwards=False, present=False))
self.last_present_forward_btn = Button(self.buttons_frm, text="PREVIOUS PRESENT", fg="blue", command=lambda: self.find_last_next_annotation(forwards=False, present=True))
self.backward_max_btn = Button(self.buttons_frm, text="<<", command=lambda: self.advance_frame(0))
self.select_frm_btn = Button(self.buttons_frm, text="Jump to selected frame", command=lambda: self.advance_frame(new_frm_number=int(self.change_frm_box.get())))
self.jump_frame = Frame(self.main_window)
self.jump = Label(self.jump_frame, text="Jump Size:")
self.jump_size = Scale(self.jump_frame, from_=0, to=100, orient=HORIZONTAL, length=200)
self.jump_size.set(0)
self.jump_back = Button(self.jump_frame, text="<<", command=lambda: self.advance_frame(int(self.change_frm_box.get()) - self.jump_size.get()))
self.jump_forward = Button(self.jump_frame, text=">>", command=lambda: self.advance_frame(int(self.change_frm_box.get()) + self.jump_size.get()))
self.folder.grid(row=0, column=1, sticky=N)
self.buttons_frm.grid(row=1, column=0)
self.change_frm_box.grid(row=0, column=1)
self.last_present_forward_btn.grid(row=1, column=0, sticky=W, padx=self.padding)
self.last_absent_forward_btn.grid(row=1, column=1, sticky=W, padx=self.padding)
self.backward_max_btn.grid(row=1, column=2, sticky=W, padx=self.padding)
self.backward_btn.grid(row=1, column=3, sticky=W, padx=self.padding)
self.change_frm_box.grid(row=1, column=4)
self.forward_btn.grid(row=1, column=5, sticky=E, padx=self.padding)
self.forward_max_btn.grid(row=1, column=6, sticky=W, padx=self.padding)
self.next_absent_forward_btn.grid(row=1, column=7, sticky=W, padx=self.padding)
self.next_present_forward_btn.grid(row=1, column=8, sticky=W, padx=self.padding)
self.select_frm_btn.grid(row=2, column=4, sticky=N)
self.jump_frame.grid(row=2, column=0)
self.jump.grid(row=0, column=0, sticky=W)
self.jump_size.grid(row=0, column=1, sticky=W)
self.jump_back.grid(row=0, column=2, sticky=E)
self.jump_forward.grid(row=0, column=3, sticky=W)
self.check_frame = Frame(self.main_window, bd=2, width=300, height=500)
self.check_frame.grid(row=0, column=1)
self.check_behavior_lbl = Label(self.check_frame, text="Check Behavior:")
self.check_behavior_lbl.config(font=("Calibri", 16))
self.check_behavior_lbl.grid(row=0, column=0, sticky=N)
self.behavior_absent_lbl = Label(self.check_frame, text="ABSENT")
self.behavior_present_lbl = Label(self.check_frame, text="PRESENT")
self.behavior_present_lbl.grid(row=1, column=1, sticky=N)
self.behavior_absent_lbl.grid(row=1, column=2, sticky=N)
self.check_present_vars, self.check_present_checkbox = {}, {}
self.check_absent_vars, self.check_absent_checkbox = {}, {}
for target_cnt, target in enumerate(self.target_lst):
self.check_present_vars[target], self.check_absent_vars[target] = (IntVar(), IntVar())
self.behavior_name_lbl = Label(self.check_frame, text=target)
self.check_present_checkbox[target] = Checkbutton(self.check_frame, variable=self.check_present_vars[target], command=lambda: self.save_behavior_in_frm(selection="present"))
self.check_absent_checkbox[target] = Checkbutton(self.check_frame, variable=self.check_absent_vars[target], command=lambda: self.save_behavior_in_frm(selection="absent"))
self.behavior_name_lbl.grid(row=target_cnt + 2, column=0, sticky=W)
self.check_present_checkbox[target].grid(row=target_cnt + 2, column=1, sticky=W)
self.check_absent_checkbox[target].grid(row=target_cnt + 2, column=2, sticky=W)
if self.data_df_targets[target].iloc[self.current_frm_n.get()] == 1:
self.check_present_vars[target].set(value=1)
elif self.data_df_targets[target].iloc[self.current_frm_n.get()] == 0:
self.check_absent_vars[target].set(value=1)
self.range_on = IntVar(value=0)
self.range_frames = Frame(self.main_window)
self.range_frames.grid(row=1, column=1, sticky=S)
self.select_range = Checkbutton(self.range_frames, text="Frame range", variable=self.range_on)
self.select_range.grid(row=0, column=0, sticky=W)
self.first_frame = Entry(self.range_frames, width=7)
self.first_frame.grid(row=0, column=1, sticky=E)
self.to_label = Label(self.range_frames, text=" to ")
self.to_label.grid(row=0, column=2, sticky=E)
self.last_frame = Entry(self.range_frames, width=7)
self.last_frame.grid(row=0, column=3, sticky=E)
save = Button(self.main_window, text="Save Range", command=lambda: self.save_behavior_in_range())
save.grid(row=2, column=1, sticky=N)
self.generate = Button(self.main_window, text="Save Annotations", command=lambda: self.save_results(), fg="blue")
self.generate.config(font=("Calibri", 16))
self.generate.grid(row=10, column=1, sticky=N)
self.video_player_frm = Frame(self.main_window, width=100, height=100)
self.video_player_frm.grid(row=0, column=2, sticky=N)
self.play_video_btn = Button(self.video_player_frm, text="Open Video", command=self.play_video)
self.play_video_btn.grid(sticky=N, pady=10)
Label(self.video_player_frm, text=self.video_presses_lbl).grid(sticky=W)
self.update_img_from_video = Button(self.video_player_frm, text="Show current video frame", command=self.update_frame_from_video)
self.update_img_from_video.grid(sticky=N)
self.bind_shortcut_keys()
Label(self.video_player_frm, text=self.key_presses_lbl).grid(sticky=S)
self.read_frm(frm_number=0)
self.main_window.mainloop()
def __create_key_presses_lbl(self):
self.key_presses_lbl = "\n\n Keyboard shortcuts for frame navigation: "
for k, v in self.img_kbd_bindings.items():
self.key_presses_lbl += '\n'
self.key_presses_lbl += v['label']
def bind_shortcut_keys(self):
self.main_window.bind(self.img_kbd_bindings['save']['kbd'], lambda x: self.save_results())
self.main_window.bind(self.img_kbd_bindings['frame+1_keep_choices']['kbd'], lambda x: self.advance_frame(new_frm_number=int(self.current_frm_n.get() + 1), keep_prior_img_cb_status=True))
self.main_window.bind(self.img_kbd_bindings['print_annotation_statistic']['kbd'], lambda x: self.__print_annotation_statistics())
self.main_window.bind(self.img_kbd_bindings['frame+1']['kbd'], lambda x: self.advance_frame(new_frm_number=int(self.current_frm_n.get() + 1)))
self.main_window.bind(self.img_kbd_bindings['frame-1']['kbd'], lambda x: self.advance_frame(new_frm_number=int(self.current_frm_n.get() - 1)))
self.main_window.bind(self.img_kbd_bindings['last_frame']['kbd'], lambda x: self.advance_frame(new_frm_number=self.max_frm_no))
self.main_window.bind(self.img_kbd_bindings['first_frame']['kbd'], lambda x: self.advance_frame(0))
def __create_video_key_presses_lbl(self):
self.video_presses_lbl = "\n\n Keyboard shortcuts for video navigation: "
for k, v in self.video_kbd_bindings.items():
self.video_presses_lbl += '\n'
self.video_presses_lbl += v['label']
def find_last_next_annotation(self, forwards: bool, present: bool):
if forwards:
sliced = self.data_df_targets.loc[self.current_frm_n.get() + 1 :,].sum(
axis=1
)
if present:
frms = list(sliced.index[sliced >= 1])
else:
frms = list(sliced.index[sliced == 0])
if len(frms) > 0:
frm = frms[0]
else:
raise FrameRangeError(
"No forwards frames with annotation detected",
source=self.__class__.__name__,
)
else:
sliced = self.data_df_targets.loc[0 : self.current_frm_n.get() - 1,].sum(
axis=1
)
if present:
frms = list(sliced.index[sliced >= 1])
else:
frms = list(sliced.index[sliced == 0])
if len(frms) > 0:
frm = frms[-1]
else:
raise FrameRangeError(
"No backwards frames with annotation detected",
source=self.__class__.__name__,
)
self.advance_frame(new_frm_number=frm, keep_prior_img_cb_status=False)
def __print_annotation_statistics(self):
table_view = [
["Video name", self.video_name],
["Video frames", self.video_meta_data["frame_count"]],
]
for target in self.target_lst:
present = sum(
self.data_df_targets[target][self.data_df_targets[target] == 1]
)
absent = sum(
self.data_df_targets[target][self.data_df_targets[target] == 0]
)
table_view.append([target + " present labels", present])
table_view.append([target + " absent labels", absent])
table_view.append(
[target + " % present", present / self.video_meta_data["frame_count"]]
)
table_view.append(
[target + " % absent", absent / self.video_meta_data["frame_count"]]
)
headers = ["VARIABLE", "VALUE"]
print(tabulate(table_view, headers, tablefmt="github"))
def play_video(self):
p = Popen(
"python {}".format(self.play_video_script_path),
stdin=PIPE,
stdout=PIPE,
shell=True,
)
main_project_dir = os.path.dirname(self.config_path)
p.stdin.write(bytes(self.video_path, "utf-8"))
p.stdin.close()
temp_file = os.path.join(main_project_dir, "subprocess.txt")
with open(temp_file, "w") as text_file:
text_file.write(str(p.pid))
def update_frame_from_video(self):
f = open(
os.path.join(os.path.dirname(self.config_path), "labelling_info.txt"), "r+"
)
os.fsync(f.fileno())
vid_frame_no = int(f.readline())
self.advance_frame(new_frm_number=vid_frame_no)
f.close()
def read_frm(self, frm_number: int):
self.cap.set(1, frm_number)
_, self.current_frm_npy = self.cap.read()
self.current_frm_npy = cv2.cvtColor(self.current_frm_npy, cv2.COLOR_RGB2BGR)
self.current_frm_pil = Image.fromarray(self.current_frm_npy)
self.current_frm_pil.thumbnail(self.max_frm_size, Image.ANTIALIAS)
self.current_frm_pil = ImageTk.PhotoImage(
master=self.main_window, image=self.current_frm_pil
)
self.video_frame = Label(self.main_window, image=self.current_frm_pil)
self.video_frame.image = self.current_frm_pil
self.video_frame.grid(row=0, column=0)
def advance_frame(
self,
new_frm_number: int,
save_behavior_in_previous_frm: bool = True,
keep_prior_img_cb_status: bool = False,
):
self.check_integrity_of_multiple_classifiers()
if new_frm_number > self.max_frm_no:
print(
"FRAME {} CANNOT BE SHOWN - YOU ARE VIEWING THE FINAL FRAME OF THE VIDEO (FRAME NUMBER {})".format(
str(new_frm_number), str(self.max_frm_no)
)
)
self.current_frm_n = IntVar(value=self.max_frm_no)
self.change_frm_box.delete(0, END)
self.change_frm_box.insert(0, self.current_frm_n.get())
elif new_frm_number < 0:
print(
"FRAME {} CANNOT BE SHOWN - YOU ARE VIEWING THE FIRST FRAME OF THE VIDEO (FRAME NUMBER {})".format(
str(new_frm_number), str(self.max_frm_no)
)
)
self.current_frm_n = IntVar(value=0)
self.change_frm_box.delete(0, END)
self.change_frm_box.insert(0, self.current_frm_n.get())
elif (new_frm_number != self.current_frm_n.get()) and (
not keep_prior_img_cb_status
):
if save_behavior_in_previous_frm:
self.save_behavior_in_frm()
self.create_print_statements()
self.current_frm_n = IntVar(value=new_frm_number)
for target in self.target_lst:
new_frame_annotation = self.data_df_targets[target].loc[
int(self.current_frm_n.get())
]
if new_frame_annotation == 0:
self.check_absent_vars[target].set(value=1)
self.check_present_vars[target].set(value=0)
elif new_frame_annotation == 1:
self.check_present_vars[target].set(value=1)
self.check_absent_vars[target].set(value=0)
else:
self.check_present_vars[target].set(value=0)
self.check_absent_vars[target].set(value=0)
self.change_frm_box.delete(0, END)
self.change_frm_box.insert(0, new_frm_number)
self.read_frm(frm_number=int(new_frm_number))
elif (new_frm_number != self.current_frm_n.get()) and (
keep_prior_img_cb_status
):
self.create_print_statements()
if save_behavior_in_previous_frm:
self.save_behavior_in_frm()
self.current_frm_n = IntVar(value=new_frm_number)
for target in self.target_lst:
new_frame_annotation = self.data_df_targets[target].loc[
int(self.current_frm_n.get() - 1)
]
if new_frame_annotation == 0:
self.check_absent_vars[target].set(value=1)
elif new_frame_annotation == 1:
self.check_present_vars[target].set(value=1)
self.change_frm_box.delete(0, END)
self.change_frm_box.insert(0, new_frm_number)
self.read_frm(frm_number=int(new_frm_number))
def save_behavior_in_frm(self, selection: str = None):
for target in self.target_lst:
target_absent_choice, target_present_choice = (
self.check_absent_vars[target].get(),
self.check_present_vars[target].get(),
)
if (target_present_choice == 1) & (selection == "present"):
self.data_df_targets[target].loc[int(self.current_frm_n.get())] = 1
self.check_absent_vars[target].set(value=0)
elif (target_absent_choice == 1) & (selection == "absent"):
self.data_df_targets[target].loc[int(self.current_frm_n.get())] = 0
self.check_present_vars[target].set(value=0)
elif (target_absent_choice == 0) & (target_present_choice == 0):
self.data_df_targets[target].loc[int(self.current_frm_n.get())] = None
elif target_present_choice == 1:
self.data_df_targets[target].loc[int(self.current_frm_n.get())] = 1
elif target_absent_choice == 1:
self.data_df_targets[target].loc[int(self.current_frm_n.get())] = 0
def save_behavior_in_range(self):
self.check_integrity_of_multiple_classifiers()
start_frm, end_frm = int(self.first_frame.get()), int(self.last_frame.get())
check_int("START FRAME", int(start_frm), max_value=self.max_frm_no, min_value=0)
check_int("END FRAME", int(end_frm), max_value=self.max_frm_no, min_value=0)
if not self.range_on.get():
raise FrameRangeError(
msg="TO SAVE RANGE OF FRAMES, TICK THE `Frame range` checkbox before clicking `Save Range`",
source=self.__class__.__name__,
)
elif start_frm < 0:
raise FrameRangeError(
msg="FRAME RANGE ERROR: START FRAME {} IS LESS THAN ZERO AND CANNOT BE SHOWN".format(
str(start_frm)
),
source=self.__class__.__name__,
)
elif end_frm > self.max_frm_no:
raise FrameRangeError(
msg="FRAME RANGE ERROR: END FRAME {} IS MORE THAN THE MAX VIDEO FRAME ({}) AND CANNOT BE SHOWN".format(
str(end_frm), str(self.max_frm_no)
),
source=self.__class__.__name__,
)
elif start_frm == end_frm:
raise FrameRangeError(
msg="FRAME RANGE ERROR: START FRAME AND END FRAME IS SET TO THE SAME VALUE ({}) AND DOES NOT REPRESENT A RANGE".format(
str(end_frm)
),
source=self.__class__.__name__,
)
elif start_frm > end_frm:
raise FrameRangeError(
msg="FRAME RANGE ERROR: START FRAME ({}) IS LARGER THAB THE END FRAME ({}). PLEASE SPECIFY A RANGE OF FRAMES WHERE THE START FRAME PRECEDE THE END FRAME".format(
str(start_frm), str(end_frm)
),
source=self.__class__.__name__,
)
else:
for frm_no in range(int(start_frm), int(end_frm) + 1):
for target in self.target_lst:
target_absent_choice, target_present_choice = (
self.check_absent_vars[target].get(),
self.check_present_vars[target].get(),
)
if target_present_choice == 1:
self.data_df_targets[target].loc[frm_no] = 1
if target_absent_choice == 1:
self.data_df_targets[target].loc[frm_no] = 0
if (target_absent_choice == 0) & (target_present_choice == 0):
self.data_df_targets[target].loc[frm_no] = None
self.read_frm(frm_number=int(end_frm))
self.change_frm_box.delete(0, END)
self.change_frm_box.insert(0, end_frm)
self.create_print_statements(
frame_range=True, start_frame=start_frm, end_frame=end_frm
)
def save_results(self):
self.save_df = read_df(self.features_extracted_file_path, self.file_type)
self.save_df = pd.concat([self.save_df, self.data_df_targets], axis=1)
self.save_df = self.save_df.dropna(subset=self.target_lst)
try:
if self.file_type == Formats.CSV.value:
self.save_df.to_csv(self.targets_inserted_file_path)
if self.file_type == Formats.PARQUET.value:
self.save_df.to_parquet(self.targets_inserted_file_path)
except Exception as e:
print(e, "SIMBA ERROR: File for video {} could not be saved.")
raise FileExistsError
stdout_success(
msg=f"SAVED: Annotation file for video {self.video_name} saved within the project_folder/csv/targets_inserted directory.",
source=self.__class__.__name__,
)
if not self.config.has_section("Last annotated frames"):
self.config.add_section("Last annotated frames")
self.config.set(
"Last annotated frames", str(self.video_name), str(self.current_frm_n.get())
)
with open(self.config_path, "w") as configfile:
self.config.write(configfile)
def check_integrity_of_multiple_classifiers(self):
none_target_lst, labelled_target_lst = [], []
for target in self.target_lst:
target_absent_choice, target_present_choice = (
self.check_absent_vars[target].get(),
self.check_present_vars[target].get(),
)
if (target_absent_choice == 0) and (target_present_choice == 0):
none_target_lst.append(target)
else:
labelled_target_lst.append(target)
if (len(none_target_lst) > 0) & (len(labelled_target_lst) > 0):
raise AdvancedLabellingError(
frame=str(self.current_frm_n.get()),
lbl_lst=labelled_target_lst,
unlabel_lst=none_target_lst,
source=self.__class__.__name__,
)
def create_print_statements(
self, frame_range: bool = None, start_frame: int = None, end_frame: int = None
):
print("USER FRAME SELECTION(S):")
if not frame_range:
for target in self.target_lst:
target_absent_choice, target_present_choice = (
self.check_absent_vars[target].get(),
self.check_present_vars[target].get(),
)
if (target_absent_choice == 1) & (target_present_choice == 0):
print(
"{} ABSENT IN FRAME {}".format(target, self.current_frm_n.get())
)
if (target_present_choice == 1) & (target_absent_choice == 0):
print(
"{} PRESENT IN FRAME {}".format(
target, self.current_frm_n.get()
)
)
if (target_present_choice == 0) & (target_absent_choice == 0):
print(
"{} UN-ANNOTATED IN FRAME {}".format(
target, self.current_frm_n.get()
)
)
if frame_range:
for target in self.target_lst:
target_absent_choice, target_present_choice = (
self.check_absent_vars[target].get(),
self.check_present_vars[target].get(),
)
if (target_present_choice == 1) & (target_absent_choice == 0):
print(
"{} PRESENT IN FRAMES {} to {}".format(
target, str(start_frame), str(end_frame)
)
)
elif (target_absent_choice == 1) & (target_present_choice == 0):
print(
"{} ABSENT IN FRAMES {} to {}".format(
target, str(start_frame), str(end_frame)
)
)
elif (target_absent_choice == 0) & (target_present_choice == 0):
print(
"{} UN-ANNOTATED IN FRAMES {} to {}".format(
target, str(start_frame), str(end_frame)
)
)
def select_labelling_video_advanced(
config_path: Union[str, os.PathLike], continuing: Optional[bool] = False
):
video_file_path = filedialog.askopenfilename(
filetypes=[("Video files", Options.ALL_VIDEO_FORMAT_STR_OPTIONS.value)]
)
check_file_exist_and_readable(video_file_path)
video_meta = get_video_meta_data(video_file_path)
_, video_name, _ = get_fn_ext(video_file_path)
print(f"ANNOTATING VIDEO {video_name} \n VIDEO INFO: {video_meta}")
_ = AdvancedLabellingInterface(
config_path=config_path, file_path=video_file_path, continuing=continuing
)
#
#
# test = select_labelling_video_advanced(config_path=r"C:\troubleshooting\two_black_animals_14bp\project_folder\project_config.ini",
# continuing=False)