Molkaatb commited on
Commit
bf4ea70
·
verified ·
1 Parent(s): 15a1d6d

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. README.md +3 -3
  2. config.yaml +75 -0
  3. model_final.pth +3 -0
README.md CHANGED
@@ -1,3 +1,3 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
1
+ # Cars Detection in Satellite Images
2
+
3
+ This is a Detectron2 Faster R-CNN model trained to detect 3 classes in satellite imagery.
config.yaml ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ "CUDNN_BENCHMARK: false\nDATALOADER:\n ASPECT_RATIO_GROUPING: true\n FILTER_EMPTY_ANNOTATIONS:\
2
+ \ true\n NUM_WORKERS: 4\n REPEAT_SQRT: true\n REPEAT_THRESHOLD: 0.0\n SAMPLER_TRAIN:\
3
+ \ TrainingSampler\nDATASETS:\n PRECOMPUTED_PROPOSAL_TOPK_TEST: 1000\n PRECOMPUTED_PROPOSAL_TOPK_TRAIN:\
4
+ \ 2000\n PROPOSAL_FILES_TEST: []\n PROPOSAL_FILES_TRAIN: []\n TEST:\n - coco_2017_val\n\
5
+ \ TRAIN:\n - coco_2017_train\nFLOAT32_PRECISION: ''\nGLOBAL:\n HACK: 1.0\nINPUT:\n\
6
+ \ CROP:\n ENABLED: false\n SIZE:\n - 0.9\n - 0.9\n TYPE: relative_range\n\
7
+ \ FORMAT: BGR\n MASK_FORMAT: polygon\n MAX_SIZE_TEST: 1333\n MAX_SIZE_TRAIN:\
8
+ \ 1333\n MIN_SIZE_TEST: 800\n MIN_SIZE_TRAIN:\n - 640\n - 672\n - 704\n -\
9
+ \ 736\n - 768\n - 800\n MIN_SIZE_TRAIN_SAMPLING: choice\n RANDOM_FLIP: horizontal\n\
10
+ MODEL:\n ANCHOR_GENERATOR:\n ANGLES:\n - - -90\n - 0\n - 90\n \
11
+ \ ASPECT_RATIOS:\n - - 0.5\n - 1.0\n - 2.0\n NAME: DefaultAnchorGenerator\n\
12
+ \ OFFSET: 0.0\n SIZES:\n - - 32\n - - 64\n - - 128\n - - 256\n\
13
+ \ - - 512\n BACKBONE:\n FREEZE_AT: 2\n NAME: build_resnet_fpn_backbone\n\
14
+ \ DEVICE: cuda\n FPN:\n FUSE_TYPE: sum\n IN_FEATURES:\n - res2\n -\
15
+ \ res3\n - res4\n - res5\n NORM: ''\n OUT_CHANNELS: 256\n KEYPOINT_ON:\
16
+ \ false\n LOAD_PROPOSALS: false\n MASK_ON: false\n META_ARCHITECTURE: GeneralizedRCNN\n\
17
+ \ PANOPTIC_FPN:\n COMBINE:\n ENABLED: true\n INSTANCES_CONFIDENCE_THRESH:\
18
+ \ 0.5\n OVERLAP_THRESH: 0.5\n STUFF_AREA_LIMIT: 4096\n INSTANCE_LOSS_WEIGHT:\
19
+ \ 1.0\n PIXEL_MEAN:\n - 103.53\n - 116.28\n - 123.675\n PIXEL_STD:\n - 1.0\n\
20
+ \ - 1.0\n - 1.0\n PROPOSAL_GENERATOR:\n MIN_SIZE: 0\n NAME: RPN\n RESNETS:\n\
21
+ \ DEFORM_MODULATED: false\n DEFORM_NUM_GROUPS: 1\n DEFORM_ON_PER_STAGE:\n\
22
+ \ - false\n - false\n - false\n - false\n DEPTH: 50\n NORM: FrozenBN\n\
23
+ \ NUM_GROUPS: 1\n OUT_FEATURES:\n - res2\n - res3\n - res4\n -\
24
+ \ res5\n RES2_OUT_CHANNELS: 256\n RES5_DILATION: 1\n STEM_OUT_CHANNELS:\
25
+ \ 64\n STRIDE_IN_1X1: true\n WIDTH_PER_GROUP: 64\n RETINANET:\n BBOX_REG_LOSS_TYPE:\
26
+ \ smooth_l1\n BBOX_REG_WEIGHTS: &id002\n - 1.0\n - 1.0\n - 1.0\n \
27
+ \ - 1.0\n FOCAL_LOSS_ALPHA: 0.25\n FOCAL_LOSS_GAMMA: 2.0\n IN_FEATURES:\n\
28
+ \ - p3\n - p4\n - p5\n - p6\n - p7\n IOU_LABELS:\n - 0\n \
29
+ \ - -1\n - 1\n IOU_THRESHOLDS:\n - 0.4\n - 0.5\n NMS_THRESH_TEST:\
30
+ \ 0.5\n NORM: ''\n NUM_CLASSES: 80\n NUM_CONVS: 4\n PRIOR_PROB: 0.01\n\
31
+ \ SCORE_THRESH_TEST: 0.05\n SMOOTH_L1_LOSS_BETA: 0.1\n TOPK_CANDIDATES_TEST:\
32
+ \ 1000\n ROI_BOX_CASCADE_HEAD:\n BBOX_REG_WEIGHTS:\n - &id001\n - 10.0\n\
33
+ \ - 10.0\n - 5.0\n - 5.0\n - - 20.0\n - 20.0\n - 10.0\n\
34
+ \ - 10.0\n - - 30.0\n - 30.0\n - 15.0\n - 15.0\n IOUS:\n\
35
+ \ - 0.5\n - 0.6\n - 0.7\n ROI_BOX_HEAD:\n BBOX_REG_LOSS_TYPE: smooth_l1\n\
36
+ \ BBOX_REG_LOSS_WEIGHT: 1.0\n BBOX_REG_WEIGHTS: *id001\n CLS_AGNOSTIC_BBOX_REG:\
37
+ \ false\n CONV_DIM: 256\n FC_DIM: 1024\n FED_LOSS_FREQ_WEIGHT_POWER: 0.5\n\
38
+ \ FED_LOSS_NUM_CLASSES: 50\n NAME: FastRCNNConvFCHead\n NORM: ''\n NUM_CONV:\
39
+ \ 0\n NUM_FC: 2\n POOLER_RESOLUTION: 7\n POOLER_SAMPLING_RATIO: 0\n \
40
+ \ POOLER_TYPE: ROIAlignV2\n SMOOTH_L1_BETA: 0.0\n TRAIN_ON_PRED_BOXES: false\n\
41
+ \ USE_FED_LOSS: false\n USE_SIGMOID_CE: false\n ROI_HEADS:\n BATCH_SIZE_PER_IMAGE:\
42
+ \ 512\n IN_FEATURES:\n - p2\n - p3\n - p4\n - p5\n IOU_LABELS:\n\
43
+ \ - 0\n - 1\n IOU_THRESHOLDS:\n - 0.5\n NAME: StandardROIHeads\n\
44
+ \ NMS_THRESH_TEST: 0.5\n NUM_CLASSES: 3\n POSITIVE_FRACTION: 0.25\n \
45
+ \ PROPOSAL_APPEND_GT: true\n SCORE_THRESH_TEST: 0.7\n ROI_KEYPOINT_HEAD:\n \
46
+ \ CONV_DIMS:\n - 512\n - 512\n - 512\n - 512\n - 512\n - 512\n\
47
+ \ - 512\n - 512\n LOSS_WEIGHT: 1.0\n MIN_KEYPOINTS_PER_IMAGE: 1\n \
48
+ \ NAME: KRCNNConvDeconvUpsampleHead\n NORMALIZE_LOSS_BY_VISIBLE_KEYPOINTS: true\n\
49
+ \ NUM_KEYPOINTS: 17\n POOLER_RESOLUTION: 14\n POOLER_SAMPLING_RATIO: 0\n\
50
+ \ POOLER_TYPE: ROIAlignV2\n ROI_MASK_HEAD:\n CLS_AGNOSTIC_MASK: false\n \
51
+ \ CONV_DIM: 256\n NAME: MaskRCNNConvUpsampleHead\n NORM: ''\n NUM_CONV:\
52
+ \ 4\n POOLER_RESOLUTION: 14\n POOLER_SAMPLING_RATIO: 0\n POOLER_TYPE: ROIAlignV2\n\
53
+ \ RPN:\n BATCH_SIZE_PER_IMAGE: 256\n BBOX_REG_LOSS_TYPE: smooth_l1\n BBOX_REG_LOSS_WEIGHT:\
54
+ \ 1.0\n BBOX_REG_WEIGHTS: *id002\n BOUNDARY_THRESH: -1\n CONV_DIMS:\n \
55
+ \ - -1\n HEAD_NAME: StandardRPNHead\n IN_FEATURES:\n - p2\n - p3\n\
56
+ \ - p4\n - p5\n - p6\n IOU_LABELS:\n - 0\n - -1\n - 1\n \
57
+ \ IOU_THRESHOLDS:\n - 0.3\n - 0.7\n LOSS_WEIGHT: 1.0\n NMS_THRESH: 0.7\n\
58
+ \ POSITIVE_FRACTION: 0.5\n POST_NMS_TOPK_TEST: 1000\n POST_NMS_TOPK_TRAIN:\
59
+ \ 1000\n PRE_NMS_TOPK_TEST: 1000\n PRE_NMS_TOPK_TRAIN: 2000\n SMOOTH_L1_BETA:\
60
+ \ 0.0\n SEM_SEG_HEAD:\n COMMON_STRIDE: 4\n CONVS_DIM: 128\n IGNORE_VALUE:\
61
+ \ 255\n IN_FEATURES:\n - p2\n - p3\n - p4\n - p5\n LOSS_WEIGHT:\
62
+ \ 1.0\n NAME: SemSegFPNHead\n NORM: GN\n NUM_CLASSES: 54\n WEIGHTS: /kaggle/working/output/model_final.pth\n\
63
+ OUTPUT_DIR: ./output\nSEED: -1\nSOLVER:\n AMP:\n ENABLED: false\n BASE_LR:\
64
+ \ 0.02\n BASE_LR_END: 0.0\n BIAS_LR_FACTOR: 1.0\n CHECKPOINT_PERIOD: 5000\n \
65
+ \ CLIP_GRADIENTS:\n CLIP_TYPE: value\n CLIP_VALUE: 1.0\n ENABLED: false\n\
66
+ \ NORM_TYPE: 2.0\n GAMMA: 0.1\n IMS_PER_BATCH: 16\n LR_SCHEDULER_NAME: WarmupMultiStepLR\n\
67
+ \ MAX_ITER: 270000\n MOMENTUM: 0.9\n NESTEROV: false\n NUM_DECAYS: 3\n REFERENCE_WORLD_SIZE:\
68
+ \ 0\n RESCALE_INTERVAL: false\n STEPS:\n - 210000\n - 250000\n WARMUP_FACTOR:\
69
+ \ 0.001\n WARMUP_ITERS: 1000\n WARMUP_METHOD: linear\n WEIGHT_DECAY: 0.0001\n\
70
+ \ WEIGHT_DECAY_BIAS: null\n WEIGHT_DECAY_NORM: 0.0\nTEST:\n AUG:\n ENABLED:\
71
+ \ false\n FLIP: true\n MAX_SIZE: 4000\n MIN_SIZES:\n - 400\n - 500\n\
72
+ \ - 600\n - 700\n - 800\n - 900\n - 1000\n - 1100\n - 1200\n\
73
+ \ DETECTIONS_PER_IMAGE: 100\n EVAL_PERIOD: 0\n EXPECTED_RESULTS: []\n KEYPOINT_OKS_SIGMAS:\
74
+ \ []\n PRECISE_BN:\n ENABLED: false\n NUM_ITER: 200\nVERSION: 2\nVIS_PERIOD:\
75
+ \ 0\n"
model_final.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:abc473bf77dbf80ca3b34e9da4f0c1212f019feb492fa1c5e0b8973ad2ff8857
3
+ size 330107220