from src.database import ModelOutputs if __name__ == "__main__": # instantiate data object annotation = { } vis_com_list = [ModelOutputs.from_annotation(path) for path in img_path_list] # generate random predictions [vis_com.generate_random_prediction() for vis_com in vis_com_list] for vis_com in vis_com_list: print(vis_com)