Spaces:
Sleeping
Sleeping
File size: 1,309 Bytes
d2ff88f |
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 |
# ---------------------------------------------------------------------------------------------------
# CLIP-DINOiser
# authors: Monika Wysoczanska, Warsaw University of Technology
# ----------------------------------------------------------------------------------------------------
# Modified from TCL
# Copyright (c) 2023 Kakao Brain. All Rights Reserved.
# ------------------------------------------------------------------------------
_base_ = ["../custom_import.py"]
# dataset settings
dataset_type = "COCOStuffDataset"
data_root = "./data/coco_stuff164k"
test_pipeline = [
dict(type="LoadImageFromFile"),
dict(type='ToRGB'),
dict(
type="MultiScaleFlipAug",
img_scale=(2048, 448),
flip=False,
transforms=[
dict(type="Resize", keep_ratio=True),
dict(type="RandomFlip"),
dict(type="ImageToTensorV2", keys=["img"]),
dict(type="Collect", keys=["img"], meta_keys=['ori_shape', 'img_shape', 'pad_shape', 'flip', 'img_info']),
],
),
]
data = dict(
test=dict(
type=dataset_type,
data_root=data_root,
img_dir="images/val2017",
ann_dir="annotations/val2017",
pipeline=test_pipeline,
)
)
test_cfg = dict(mode="slide", stride=(224, 224), crop_size=(448, 448))
|