Upload folder using huggingface_hub
Browse files- README.md +3 -3
- config.yaml +75 -0
- model_final.pth +3 -0
README.md
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
-
|
2 |
-
|
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
|