Skip to content
This repository has been archived by the owner on Mar 19, 2024. It is now read-only.

Commit

Permalink
Add geolocalization configs for models
Browse files Browse the repository at this point in the history 
Summary: as title

Reviewed By: QuentinDuval

Differential Revision: D33767598

fbshipit-source-id: 4708b7a471077a238bf32523356892d1fdba7dc5
  • Loading branch information
@prigoyal @facebook-github-bot
prigoyal authored and facebook-github-bot committed Jan 31, 2022
1 parent 9459d00 commit 8c5ba7b
Show file tree
Hide file tree
Showing 18 changed files with 253 additions and 0 deletions.
8 changes: 8 additions & 0 deletions configs/config/benchmark/geolocalization/dataset/im2gps237_test.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# @package _global_
config:
DATA:
TEST:
DATA_SOURCES: [disk_filelist]
LABEL_SOURCES: []
LABEL_TYPE: sample_index
DATASET_NAMES: [im2gps_237_test_filelist]
8 changes: 8 additions & 0 deletions configs/config/benchmark/geolocalization/dataset/im2gps3k_test.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# @package _global_
config:
DATA:
TEST:
DATA_SOURCES: [disk_filelist]
LABEL_SOURCES: []
LABEL_TYPE: sample_index
DATASET_NAMES: [im2gps_3k_test_filelist]
45 changes: 45 additions & 0 deletions configs/config/benchmark/geolocalization/eval_resnet_1gpu_geolocalization.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,45 @@
# @package _global_
engine_name: extract_label_predictions
config:
TEST_ONLY: True
TEST_MODEL: True
METERS:
name: accuracy_list_meter
accuracy_list_meter:
num_meters: 1
topk_values: [1]
DATA:
NUM_DATALOADER_WORKERS: 5
TEST:
DATA_SOURCES: [disk_filelist]
LABEL_SOURCES: []
LABEL_TYPE: sample_index
DATASET_NAMES: [im2gps_3k_test_filelist]
BATCHSIZE_PER_REPLICA: 32
TRANSFORMS:
- name: Resize
size: 224
- name: CenterCrop
size: 224
- name: ToTensor
- name: Normalize
mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225]
MMAP_MODE: False
MODEL:
TRUNK:
NAME: resnet
TRUNK_PARAMS:
RESNETS:
DEPTH: 50
WEIGHTS_INIT:
PARAMS_FILE: "specify the model weights"
STATE_DICT_KEY_NAME: classy_state_dict
DISTRIBUTED:
NUM_NODES: 1
NUM_PROC_PER_NODE: 1
MACHINE:
DEVICE: gpu
GEO_LOCALIZATION:
TRAIN_LABEL_MAPPING: "manifold://ssl_framework/tree/datasets/im2gps_parsed_data/mp16_places365_mapping_h3.json"
ACC_KM_THRESHOLDS: [1, 25, 200, 750, 2500]
13 changes: 13 additions & 0 deletions configs/config/benchmark/geolocalization/models_coarse/regnet128gf.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
# @package _global_
config:
MODEL:
FEATURE_EVAL_SETTINGS:
EVAL_MODE_ON: True
FREEZE_TRUNK_AND_HEAD: True
EVAL_TRUNK_AND_HEAD: True
TRUNK:
NAME: regnet
REGNET:
name: regnet_y_128gf
HEAD:
PARAMS: [['mlp', {'dims': [7392, 3298]}]]
17 changes: 17 additions & 0 deletions configs/config/benchmark/geolocalization/models_coarse/regnet256gf_1.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
# @package _global_
config:
MODEL:
FEATURE_EVAL_SETTINGS:
EVAL_MODE_ON: True
FREEZE_TRUNK_AND_HEAD: True
EVAL_TRUNK_AND_HEAD: True
TRUNK:
NAME: regnet
REGNET:
depth: 27
w_0: 640
w_a: 230.83
w_m: 2.53
group_width: 373
HEAD:
PARAMS: [['mlp', {'dims': [10444, 3298]}]]
17 changes: 17 additions & 0 deletions configs/config/benchmark/geolocalization/models_coarse/regnet64gf_3.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
# @package _global_
config:
MODEL:
FEATURE_EVAL_SETTINGS:
EVAL_MODE_ON: True
FREEZE_TRUNK_AND_HEAD: True
EVAL_TRUNK_AND_HEAD: True
TRUNK:
NAME: regnet
REGNET:
depth: 20
w_0: 352
w_a: 147.48
w_m: 2.4
group_width: 328
HEAD:
PARAMS: [['mlp', {'dims': [4920, 3298]}]]
13 changes: 13 additions & 0 deletions configs/config/benchmark/geolocalization/models_coarse/regnet8gf.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
# @package _global_
config:
MODEL:
FEATURE_EVAL_SETTINGS:
EVAL_MODE_ON: True
FREEZE_TRUNK_AND_HEAD: True
EVAL_TRUNK_AND_HEAD: True
TRUNK:
NAME: regnet
REGNET:
name: regnet_y_8gf
HEAD:
PARAMS: [['mlp', {'dims': [2016, 3298]}]]
13 changes: 13 additions & 0 deletions configs/config/benchmark/geolocalization/models_fine/regnet128gf.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
# @package _global_
config:
MODEL:
FEATURE_EVAL_SETTINGS:
EVAL_MODE_ON: True
FREEZE_TRUNK_AND_HEAD: True
EVAL_TRUNK_AND_HEAD: True
TRUNK:
NAME: regnet
REGNET:
name: regnet_y_128gf
HEAD:
PARAMS: [['mlp', {'dims': [7392, 12893]}]]
17 changes: 17 additions & 0 deletions configs/config/benchmark/geolocalization/models_fine/regnet256gf_1.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
# @package _global_
config:
MODEL:
FEATURE_EVAL_SETTINGS:
EVAL_MODE_ON: True
FREEZE_TRUNK_AND_HEAD: True
EVAL_TRUNK_AND_HEAD: True
TRUNK:
NAME: regnet
REGNET:
depth: 27
w_0: 640
w_a: 230.83
w_m: 2.53
group_width: 373
HEAD:
PARAMS: [['mlp', {'dims': [10444, 12893]}]]
17 changes: 17 additions & 0 deletions configs/config/benchmark/geolocalization/models_fine/regnet64gf_3.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
# @package _global_
config:
MODEL:
FEATURE_EVAL_SETTINGS:
EVAL_MODE_ON: True
FREEZE_TRUNK_AND_HEAD: True
EVAL_TRUNK_AND_HEAD: True
TRUNK:
NAME: regnet
REGNET:
depth: 20
w_0: 352
w_a: 147.48
w_m: 2.4
group_width: 328
HEAD:
PARAMS: [['mlp', {'dims': [4920, 12893]}]]
13 changes: 13 additions & 0 deletions configs/config/benchmark/geolocalization/models_fine/regnet8gf.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
# @package _global_
config:
MODEL:
FEATURE_EVAL_SETTINGS:
EVAL_MODE_ON: True
FREEZE_TRUNK_AND_HEAD: True
EVAL_TRUNK_AND_HEAD: True
TRUNK:
NAME: regnet
REGNET:
name: regnet_y_8gf
HEAD:
PARAMS: [['mlp', {'dims': [2016, 12893]}]]
13 changes: 13 additions & 0 deletions configs/config/benchmark/geolocalization/models_middle/regnet128gf.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
# @package _global_
config:
MODEL:
FEATURE_EVAL_SETTINGS:
EVAL_MODE_ON: True
FREEZE_TRUNK_AND_HEAD: True
EVAL_TRUNK_AND_HEAD: True
TRUNK:
NAME: regnet
REGNET:
name: regnet_y_128gf
HEAD:
PARAMS: [['mlp', {'dims': [7392, 7202]}]]
17 changes: 17 additions & 0 deletions configs/config/benchmark/geolocalization/models_middle/regnet256gf_1.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
# @package _global_
config:
MODEL:
FEATURE_EVAL_SETTINGS:
EVAL_MODE_ON: True
FREEZE_TRUNK_AND_HEAD: True
EVAL_TRUNK_AND_HEAD: True
TRUNK:
NAME: regnet
REGNET:
depth: 27
w_0: 640
w_a: 230.83
w_m: 2.53
group_width: 373
HEAD:
PARAMS: [['mlp', {'dims': [10444, 7202]}]]
17 changes: 17 additions & 0 deletions configs/config/benchmark/geolocalization/models_middle/regnet64gf_3.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
# @package _global_
config:
MODEL:
FEATURE_EVAL_SETTINGS:
EVAL_MODE_ON: True
FREEZE_TRUNK_AND_HEAD: True
EVAL_TRUNK_AND_HEAD: True
TRUNK:
NAME: regnet
REGNET:
depth: 20
w_0: 352
w_a: 147.48
w_m: 2.4
group_width: 328
HEAD:
PARAMS: [['mlp', {'dims': [4920, 7202]}]]
13 changes: 13 additions & 0 deletions configs/config/benchmark/geolocalization/models_middle/regnet8gf.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
# @package _global_
config:
MODEL:
FEATURE_EVAL_SETTINGS:
EVAL_MODE_ON: True
FREEZE_TRUNK_AND_HEAD: True
EVAL_TRUNK_AND_HEAD: True
TRUNK:
NAME: regnet
REGNET:
name: regnet_y_8gf
HEAD:
PARAMS: [['mlp', {'dims': [2016, 7202]}]]
4 changes: 4 additions & 0 deletions configs/config/benchmark/geolocalization/partitioning/coarse.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
# @package _global_
config:
GEO_LOCALIZATION:
TRAIN_LABEL_MAPPING: "manifold://ssl_framework/tree/datasets/im2gps_parsed_data/s2_cells/cells_50_5000.csv"
4 changes: 4 additions & 0 deletions configs/config/benchmark/geolocalization/partitioning/fine.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
# @package _global_
config:
GEO_LOCALIZATION:
TRAIN_LABEL_MAPPING: "manifold://ssl_framework/tree/datasets/im2gps_parsed_data/s2_cells/cells_50_1000.csv"
4 changes: 4 additions & 0 deletions configs/config/benchmark/geolocalization/partitioning/middle.yaml
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
# @package _global_
config:
GEO_LOCALIZATION:
TRAIN_LABEL_MAPPING: "manifold://ssl_framework/tree/datasets/im2gps_parsed_data/s2_cells/cells_50_2000.csv"

0 comments on commit 8c5ba7b

Please sign in to comment.