cfg.embed_size, cfg.batch_size, cfg.lr, cfg.step_size, cfg.cap_embed_type, cfg.image_melt_layer, cfg.np) if cfg.note: cfg.exp_name += "_" + cfg.note cfg.model_path = os.path.join("/shared/rsaas/aiyucui2/wider_person", cfg.model_path, cfg.exp_name) cfg.output_path = os.path.join("/shared/rsaas/aiyucui2/wider_person", cfg.output_path, cfg.exp_name) if not os.path.exists(cfg.model_path): os.mkdir(cfg.model_path) if not os.path.exists(cfg.output_path): os.mkdir(cfg.output_path) logger = Logger(os.path.join(cfg.output_path, "log.txt")) logger.log(str(cfg)) #------------------ ## agents #------------------ # Data loaders train_loader = build_wider_dataloader(cfg) cfg.num_ids = len(train_loader.dataset.person2label.values()) test_text_loader = build_text_test_loader(cfg) test_image_loader = build_image_test_loader(cfg) # Evaluators Evaluator = NPEvaluator if cfg.np else GlobalEvaluator evaluator = Evaluator(img_loader=test_image_loader, cap_loader=test_text_loader,
cfg.lr, cfg.cap_embed_type, cfg.image_melt_layer, cfg.cos_margin, cfg.np) cfg.exp_name += "_id_only" cfg.model_path = os.path.join("/shared/rsaas/aiyucui2/wider_person", cfg.model_path, cfg.exp_name) cfg.output_path = os.path.join("/shared/rsaas/aiyucui2/wider_person", cfg.output_path, cfg.exp_name) if not os.path.exists(cfg.model_path): os.mkdir(cfg.model_path) if not os.path.exists(cfg.output_path): os.mkdir(cfg.output_path) logger = Logger(os.path.join(cfg.output_path, "log.txt")) logger.log(str(cfg)) #------------------ ## agents #------------------ # Data loaders train_loader = build_wider_dataloader(cfg) cfg.num_ids = len(train_loader.dataset.person2label.values()) test_text_loader = build_text_test_loader(cfg) test_image_loader = build_image_test_loader(cfg) # Evaluators Evaluator = NPEvaluator if cfg.np else GlobalEvaluator evaluator = Evaluator(img_loader=test_image_loader,