-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathbase.yaml
45 lines (45 loc) · 1.44 KB
/
base.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
VIDEO:
FILE_NAME: "examples/VID_altes_Hauptgebaeude.mp4" # either path to a video, or 0 for webcam
MASKS_PATH: "masks/"
OVERLAYS_PATH: "masks/overlay/"
FEATURE_TYPE: "SIFT"
MODEL_CHECKPOINT: "mobile-netv2.pt"
DOWNLOAD: True # Set to False, in case you want to use an offline model
ANNOTATIONS_PATH: "annotations.json"
DEVICE: "cuda"
TRAINING:
# One of "fasterrcnn_resnet50_fpn", "fasterrcnn_mobilenet_v3_large_fpn"
META_ARCHITECTURE: "fasterrcnn_mobilenet_v3_large_fpn"
# One of "FasterRCNN_ResNet50_FPN_Weights", "FasterRCNN_MobileNet_V3_Large_FPN_Weights"
WEIGHTS: "FasterRCNN_MobileNet_V3_Large_FPN_Weights"
MODEL_FILE: "mobile-netv2.pt"
NUM_CLASSES: 17
TEST_PROPORTION: 0.2
PARAMETERS:
BASE_LR: 0.001
MOMENTUM: 0.9
WEIGHT_DECAY: 0.0001
BATCH_SIZE: 4
EPOCHS: 40
STEPS: 20 # StepLR only accepts a single number
GAMMA: 0.1
NUM_WORKERS: 4
DEVICE: "cuda"
EVALUATION:
MODEL_CHECKPOINT: "mobile-netv2.pt"
DOWNLOAD: True # Set to False, in case you want to use an offline model
BATCH_SIZE: 1
NUM_WORKERS: 2
DEVICE: "cuda"
DATA:
PATH: "data"
ANNOTATIONS_PATH: "annotations.json"
SUPERCATEGORIES: ["hauptgebaeude", "karo5", "piloty", "ULB"]
SUBCATEGORIES: ["right", "back", "left", "front"]
METHOD:
NAME: "execute" # one of execute, train, view, label
MODE: "annotate" # if NAME=="label", either "annotate" or "review"
WANDB:
ENTITY: "macaw"
PROJECT: "augmented-vision"
IGNORE: True