add files

Author: Spritea

.gitignore

@@ -0,0 +1,18 @@
+# img
+*.bmp
+*.jpg
+*.png
+*.tif
+*.tiff
+# numpy
+*.npy
+# py buffer
+*.pyc
+# pycharm
+.idea
+/.idea/
+# pytorch
+*.pt
+/runs/
+/pretrain/
+/dataset/

configs/fcn8s_pascal.yml

@@ -0,0 +1,26 @@
+model:
+    arch: fcn8s
+data:
+    dataset: pascal
+    train_split: train_aug
+    val_split: val
+    img_rows: 'same'
+    img_cols: 'same'
+    path: /private/home/meetshah/datasets/VOC/060817/VOCdevkit/VOC2012/
+    sbd_path: /private/home/meetshah/datasets/VOC/benchmark_RELEASE/
+training:
+    train_iters: 300000
+    batch_size: 1
+    val_interval: 1000
+    n_workers: 16
+    print_interval: 50
+    optimizer:
+        name: 'sgd'
+        lr: 1.0e-10
+        weight_decay: 0.0005
+        momentum: 0.99
+    loss:
+        name: 'cross_entropy'
+        size_average: False
+    lr_schedule:
+    resume: fcn8s_pascal_best_model.pkl

configs/frrnB_cityscapes.yml

@@ -0,0 +1,22 @@
+model:
+    arch: frrnB
+data:
+    dataset: cityscapes
+    train_split: train
+    val_split: val
+    img_rows: 512
+    img_cols: 1024
+    path: /private/home/meetshah/misc_code/ps/data/VOCdevkit/VOC2012/
+training:
+    train_iters: 85000
+    batch_size: 2
+    val_interval: 500
+    print_interval: 25
+    optimizer:
+        lr: 1.0e-4
+    l_rate: 1.0e-4
+    l_schedule:
+    momentum: 0.99
+    weight_decay: 0.0005
+    resume: frrnB_cityscapes_best_model.pkl
+    visdom: False

ptsemseg/__init__.py

@@ -0,0 +1,49 @@
+import logging
+from ptsemseg.augmentations.augmentations import (
+    AdjustContrast,
+    AdjustGamma,
+    AdjustBrightness,
+    AdjustSaturation,
+    AdjustHue,
+    RandomCrop,
+    RandomHorizontallyFlip,
+    RandomVerticallyFlip,
+    Scale,
+    RandomSized,
+    RandomSizedCrop,
+    RandomRotate,
+    RandomTranslate,
+    CenterCrop,
+    Compose,
+)
+
+logger = logging.getLogger("ptsemseg")
+
+key2aug = {
+    "gamma": AdjustGamma,
+    "hue": AdjustHue,
+    "brightness": AdjustBrightness,
+    "saturation": AdjustSaturation,
+    "contrast": AdjustContrast,
+    "rcrop": RandomCrop,
+    "hflip": RandomHorizontallyFlip,
+    "vflip": RandomVerticallyFlip,
+    "scale": Scale,
+    "rsize": RandomSized,
+    "rsizecrop": RandomSizedCrop,
+    "rotate": RandomRotate,
+    "translate": RandomTranslate,
+    "ccrop": CenterCrop,
+}
+
+
+def get_composed_augmentations(aug_dict):
+    if aug_dict is None:
+        logger.info("Using No Augmentations")
+        return None
+
+    augmentations = []
+    for aug_key, aug_param in aug_dict.items():
+        augmentations.append(key2aug[aug_key](aug_param))
+        logger.info("Using {} aug with params {}".format(aug_key, aug_param))
+    return Compose(augmentations)