zson
zson copied to clipboard
How to use the extract-goal-features.py ?
Thanks for your great work ,I want to reproduce the work and re-train the model.but I do not know what does the 'chunk_index' mean in the extract-goal-features.py? what value should I give to the 'chunk_index' when run the extract-goal-features.py file. As follows: parser.add_argument( "--chunk_index", required=True, type=int, help="the chunk index", )
python tools/extract-goal-features.py --chunks 1 --chunk_index 0 --clip_model RN50 --dataset hm3d --split train
I am using a single graphics card A800, without the need for partitioning. In the end, 800. npy files were generated. Please note that there are two versions, 2 * 800. npy files, and the following comments need to be removed from the tools/extract-goal-features.py file
VERSION = "v1"
IMG_WIDTH = 512
IMG_HEIGHT = 512
SENSOR_HFOV = 90
SENSOR_HEIGHT = 1.25
DATASET_VERSION = "v1"
# VERSION = "v2"
# IMG_WIDTH = 640
# IMG_HEIGHT = 480
# SENSOR_HFOV = 90
# SENSOR_HEIGHT = 0.88
# DATASET_VERSION = "v2"