Esempio n. 1
0
    # config_FINN_W2_A2_28.examplar_size = 130
    # config_FINN_W2_A2_28.instance_size = 256
    # config_FINN_W2_A2_28.score_size = 30
    # save_config(config_FINN_W2_A2_28, f"./models_{exp_nbr}/")
    # # training SiamFC network, using GPU by default
    # train(data_dir, train_imdb, val_imdb, model_save_path=f"./models_{exp_nbr}/",
    #       type=NetType.FINN_W2_A2_X28,
    #       config=config_FINN_W2_A2_28)
    # writer.close()

    exp_nbr = "FINN_W2_A2_29" + "_" + str(dt.now())
    writer = SummaryWriter(f"runs/{exp_nbr}")
    config_FINN_W2_A2_29 = Config()
    config_FINN_W2_A2_29.val_videos = 400
    config_FINN_W2_A2_29.examplar_size = 130
    config_FINN_W2_A2_29.instance_size = 256
    config_FINN_W2_A2_29.score_size = 30
    save_config(config_FINN_W2_A2_29, f"./models_{exp_nbr}/")
    # training SiamFC network, using GPU by default
    train(data_dir,
          train_imdb,
          val_imdb,
          model_save_path=f"./models_{exp_nbr}/",
          type=NetType.FINN_W2_A2_X29,
          config=config_FINN_W2_A2_29)
    writer.close()

    exp_nbr = "FINN_W2_A2_30" + "_" + str(dt.now())
    writer = SummaryWriter(f"runs/{exp_nbr}")
    config_FINN_W2_A2_30 = Config()
    config_FINN_W2_A2_30.val_videos = 400
    # exp_nbr = "FINN_W2_A8_X27_z256_x130" + "_" + str(dt.now())
    # writer = SummaryWriter(f"runs/{exp_nbr}")
    # config_FINN_W2_A8_X27_z256_x130 = Config()
    # config_FINN_W2_A8_X27_z256_x130.val_videos = 400
    # config_FINN_W2_A8_X27_z256_x130.examplar_size = 130
    # config_FINN_W2_A8_X27_z256_x130.instance_size = 256
    # config_FINN_W2_A8_X27_z256_x130.score_size = 30
    # save_config(config_FINN_W2_A8_X27_z256_x130, f"./models_{exp_nbr}/")
    # # training SiamFC network, using GPU by default
    # train(data_dir, train_imdb, val_imdb, model_save_path=f"./models_{exp_nbr}/",
    #       type=NetType.FINN_W2_A8_X27_z256_x130,
    #       config=config_FINN_W2_A8_X27_z256_x130)
    # writer.close()

    exp_nbr = "FINN_W2_A2_X26_z256_x130" + "_" + str(dt.now())
    writer = SummaryWriter(f"runs/{exp_nbr}")
    config_FINN_W2_A2_X26_z256_x130 = Config()
    config_FINN_W2_A2_X26_z256_x130.val_videos = 400
    config_FINN_W2_A2_X26_z256_x130.examplar_size = 130
    config_FINN_W2_A2_X26_z256_x130.instance_size = 256
    config_FINN_W2_A2_X26_z256_x130.score_size = 30
    save_config(config_FINN_W2_A2_X26_z256_x130, f"./models_{exp_nbr}/")
    # training SiamFC network, using GPU by default
    train(data_dir,
          train_imdb,
          val_imdb,
          model_save_path=f"./models_{exp_nbr}/",
          type=NetType.FINN_W2_A2_X26_z256_x130,
          config=config_FINN_W2_A2_X26_z256_x130)
    writer.close()