def make_arg_parser():
parser = argparse.ArgumentParser()
ImageFeatures.add_args(parser)
parser.add_argument("--load_scorer", type=str, default='')
parser.add_argument("--load_follower", type=str, default='')
parser.add_argument("--load_traj_encoder", type=str, default='')
parser.add_argument( "--feedback_method",
choices=["sample", "teacher", "sample1step","sample2step","sample3step","teacher+sample","recover"], default="sample")
parser.add_argument("--debug", action='store_true')
parser.add_argument("--bidirectional", action='store_true')
parser.add_argument("--transformer", action='store_true')
parser.add_argument("--scorer", action='store_true')
parser.add_argument("--coground", action='store_false')
parser.add_argument("--prog_monitor", action='store_false')
parser.add_argument("--dev_monitor", action='store_true')
parser.add_argument("--bt_button", action='store_true')
parser.add_argument("--soft_align", action='store_true')
parser.add_argument("--n_iters", type=int, default=20000)
parser.add_argument("--num_head", type=int, default=1)
parser.add_argument("--use_pretraining", action='store_true')
parser.add_argument("--grad", type=str, default='all')
parser.add_argument("--pretrain_splits", nargs="+", default=[])
parser.add_argument("--n_pretrain_iters", type=int, default=50000)
parser.add_argument("--no_save", action='store_true')
parser.add_argument("--use_glove", action='store_true')
parser.add_argument("--attn_only_verb", action='store_true')
parser.add_argument("--use_train_subset", action='store_true',
help="use a subset of the original train data for validation")
parser.add_argument("--use_test_set", action='store_true')
parser.add_argument("--seed", type=int, default=1)
return parser
def make_arg_parser():
parser = argparse.ArgumentParser()
ImageFeatures.add_args(parser)
parser.add_argument(
"--use_train_subset", action='store_true',
help="use a subset of the original train data for validation")
parser.add_argument("--n_iters", type=int, default=20000)
parser.add_argument("--no_save", action='store_true')
return parser
def make_arg_parser():
parser = argparse.ArgumentParser()
ImageFeatures.add_args(parser)
parser.add_argument(
"--use_train_subset",
action='store_true',
help="use a subset of the original train data for validation")
parser.add_argument("--n_iters", type=int, default=20000)
parser.add_argument("--no_save", action='store_true')
parser.add_argument("--result_dir", default=RESULT_DIR)
parser.add_argument("--snapshot_dir", default=SNAPSHOT_DIR)
parser.add_argument("--plot_dir", default=PLOT_DIR)
return parser
def make_arg_parser():
parser = argparse.ArgumentParser()
ImageFeatures.add_args(parser)
parser.add_argument(
"--feedback_method", choices=["sample", "teacher", "teacher+sample"],
default="sample")
parser.add_argument("--bidirectional", action='store_true')
parser.add_argument("--n_iters", type=int, default=20000)
parser.add_argument("--use_pretraining", action='store_true')
parser.add_argument("--pretrain_splits", nargs="+", default=[])
parser.add_argument("--n_pretrain_iters", type=int, default=50000)
parser.add_argument("--no_save", action='store_true')
parser.add_argument(
"--use_train_subset", action='store_true',
help="use a subset of the original train data for validation")
parser.add_argument("--use_test_set", action='store_true')
return parser
import torch
import env
import numpy as np
import json
import sys
sys.path.append('build')
import MatterSim
import math
from vocab import SUBTRAIN_VOCAB, TRAIN_VOCAB, TRAINVAL_VOCAB
import argparse
parser = argparse.ArgumentParser()
from env import ImageFeatures
ImageFeatures.add_args(parser)
import utils
from utils import read_vocab, Tokenizer, timeSince, try_cuda, vocab_pad_idx
from env import R2RBatch, ImageFeatures
from vocab import SUBTRAIN_VOCAB, TRAIN_VOCAB, TRAINVAL_VOCAB
from model import CompatLanEncoderLSTM, CompatVisEncoderLSTM, dotSimilarity, EncoderLSTM,SpeakerEncoderLSTM
from compatModel import compatModel
args, _ = parser.parse_known_args()
image_features_list= ImageFeatures.from_args(args)
angle_inc = np.pi / 6.
def build_viewpoint_loc_embedding(viewIndex):
"""
Position embedding:
def make_arg_parser():
parser = argparse.ArgumentParser()
ImageFeatures.add_args(parser)
parser.add_argument("--load_scorer", type=str, default='')
parser.add_argument("--load_follower", type=str, default='')
parser.add_argument("--load_traj_encoder", type=str, default='')
parser.add_argument("--feedback_method",
choices=[
"sample", "teacher", "sample1step", "sample2step",
"sample3step", "teacher+sample", "recover"
],
default="sample")
parser.add_argument("--debug", action='store_true')
parser.add_argument("--bidirectional", action='store_true')
parser.add_argument("--transformer", action='store_true')
parser.add_argument("--scorer", action='store_true')
parser.add_argument("--coground", action='store_false')
parser.add_argument("--prog_monitor", action='store_false')
parser.add_argument("--dev_monitor", action='store_true')
parser.add_argument("--bt_button", action='store_true')
parser.add_argument("--soft_align", action='store_true')
parser.add_argument("--n_iters", type=int, default=10900)
parser.add_argument("--num_head", type=int, default=1)
parser.add_argument("--use_pretraining", action='store_true')
parser.add_argument("--grad", type=str, default='all')
parser.add_argument("--pretrain_splits", nargs="+", default=[])
parser.add_argument("--n_pretrain_iters", type=int, default=50000)
parser.add_argument("--no_save", action='store_true')
parser.add_argument("--use_glove", action='store_true')
parser.add_argument("--WIDTH", type=int, default=640)
parser.add_argument("--HEIGHT", type=int, default=480)
parser.add_argument("--VFOV", type=int, default=60)
parser.add_argument("--useStopFeat", type=int, default=1)
parser.add_argument("--useObjLabelOrVis",
type=str,
default='both',
help="options: vis, label, both, none")
parser.add_argument("--objFeatType",
type=str,
default='fc7',
help="options: pool5, fc7")
parser.add_argument("--objVisFeatDim", type=int, default=2048)
parser.add_argument("--objLanFeatDim", type=int, default=512)
parser.add_argument("--objTopK", type=int, default=3)
parser.add_argument("--useDect", type=bool, default=False)
parser.add_argument("--instrType",
type=str,
default='instructions',
help="options: instrutions, instructions_l")
parser.add_argument("--ObjEachViewVisFeatDir",
type=str,
default='objEachViewVisFeatFc7Top3/')
parser.add_argument("--ObjEachViewLanFeatDir",
type=str,
default='objEachViewLanFeatTop3/')
parser.add_argument(
"--labelGlovePath",
type=str, # for object label embedding
default='tasks/REVERIE/data/reverie4_reverie4.npy')
parser.add_argument(
"--grdModelPrefix",
type=str, # for visual grounding
default='MAttNet3/output/reverie4_reverie4/mrcn_cmr_with_st')
parser.add_argument("--matterportDir",
type=str,
default='/home/qyk/dataset/Matterport/')
parser.add_argument("--attn_only_verb", action='store_true')
parser.add_argument(
"--use_train_subset",
action='store_true',
help="use a subset of the original train data for validation")
parser.add_argument("--use_test_set", action='store_true')
parser.add_argument("--seed", type=int, default=1)
return parser
def make_arg_parser():
parser = argparse.ArgumentParser()
ImageFeatures.add_args(parser)
Refer360ImageFeatures.add_args(parser)
parser.add_argument(
"--use_train_subset",
action='store_true',
help="use a subset of the original train data for validation")
parser.add_argument("--bidirectional", action='store_true')
parser.add_argument("--word_embedding_size", type=int, default=300)
#parser.add_argument("--hidden_size", type=int, default=512)
parser.add_argument("--hidden_size", type=int, default=256)
parser.add_argument("--learning_rate", type=float, default=0.0001)
parser.add_argument("--weight_decay", type=float, default=0.0005)
parser.add_argument("--dropout_ratio", type=float, default=0.5)
parser.add_argument("--feedback_method",
choices=['teacher', 'sample'],
default='teacher')
parser.add_argument("--n_iters", type=int, default=100000)
parser.add_argument("--log_every", type=int, default=5000)
parser.add_argument("--save_every", type=int, default=5000)
parser.add_argument("--max_input_length", type=int, default=80)
parser.add_argument("--seed", type=int, default=10)
parser.add_argument("--beam_size", type=int, default=1)
parser.add_argument("--no_save", action='store_true')
parser.add_argument("--prefix", type=str, default='R2R')
parser.add_argument("--language", type=str, default='en-ALL')
parser.add_argument('--wordvec_path',
type=str,
default='tasks/R2R/data/train_glove')
parser.add_argument('--wordvec_finetune', action='store_true')
parser.add_argument("--error_margin", type=float, default=3.0)
parser.add_argument("--use_intermediate", action='store_true')
parser.add_argument('--use_reading', action='store_true')
parser.add_argument('--use_raw', action='store_true')
parser.add_argument("--add_asterix", action='store_true')
#parser.add_argument("--env", type=str, default='r2r')
parser.add_argument('--img_features_root',
type=str,
default='./img_features')
parser.add_argument(
'--cache_root',
type=str,
default='/projects/vcirik/refer360/data/cached_data_15degrees/')
parser.add_argument('--image_list_file',
type=str,
default='/projects/vcirik/refer360/data/imagelist.txt')
parser.add_argument(
'--refer360_root',
type=str,
default='/projects/vcirik/refer360/data/continuous_grounding')
parser.add_argument("--angle_inc", type=int, default=30)
parser.add_argument('--use_gt_actions', action='store_true')
parser.add_argument('--use_absolute_location_embeddings',
action='store_true')
parser.add_argument('--use_stop_embeddings', action='store_true')
parser.add_argument('--use_timestep_embeddings', action='store_true')
parser.add_argument('--use_visited_embeddings',
type=str,
choices=['', 'ones', 'zeros', 'count', 'pe'],
default='')
parser.add_argument('--use_oracle_embeddings', action='store_true')
parser.add_argument('--use_object_embeddings', action='store_true')
parser.add_argument('--metrics',
type=str,
default='success',
help='Success metric, default=success')
parser.add_argument('--deaf', action='store_true')
parser.add_argument('--blind', action='store_true')
parser.add_argument('--no_lookahead', action='store_true')
parser.add_argument('--nextstep', action='store_true')
parser.add_argument("--verbose", action='store_true')
return parser