zson icon indicating copy to clipboard operation
zson copied to clipboard

How to use the extract-goal-features.py ?

Open Sky-railway opened this issue 2 years ago • 1 comments

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", )

Sky-railway avatar Apr 21 '24 08:04 Sky-railway

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"

Lazy-boy193 avatar Apr 28 '25 03:04 Lazy-boy193