12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394 |
- import paddlers as pdrs
- from paddlers import transforms as T
- DATA_DIR = './data/airchange/'
- TRAIN_FILE_LIST_PATH = './data/airchange/train.txt'
- EVAL_FILE_LIST_PATH = './data/airchange/eval.txt'
- EXP_DIR = './output/bit/'
- pdrs.utils.download_and_decompress(
- 'https://paddlers.bj.bcebos.com/datasets/airchange.zip', path='./data/')
- train_transforms = T.Compose([
-
- T.DecodeImg(),
-
- T.RandomCrop(
-
- crop_size=256,
-
- aspect_ratio=[0.5, 2.0],
-
- scaling=[0.2, 1.0]),
-
- T.RandomHorizontalFlip(prob=0.5),
-
- T.Normalize(
- mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5]),
- T.ArrangeChangeDetector('train')
- ])
- eval_transforms = T.Compose([
- T.DecodeImg(),
-
- T.Normalize(
- mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5]),
- T.ReloadMask(),
- T.ArrangeChangeDetector('eval')
- ])
- train_dataset = pdrs.datasets.CDDataset(
- data_dir=DATA_DIR,
- file_list=TRAIN_FILE_LIST_PATH,
- label_list=None,
- transforms=train_transforms,
- num_workers=0,
- shuffle=True,
- with_seg_labels=False,
- binarize_labels=True)
- eval_dataset = pdrs.datasets.CDDataset(
- data_dir=DATA_DIR,
- file_list=EVAL_FILE_LIST_PATH,
- label_list=None,
- transforms=eval_transforms,
- num_workers=0,
- shuffle=False,
- with_seg_labels=False,
- binarize_labels=True)
- model = pdrs.tasks.cd.BIT()
- model.train(
- num_epochs=10,
- train_dataset=train_dataset,
- train_batch_size=4,
- eval_dataset=eval_dataset,
- save_interval_epochs=3,
-
- log_interval_steps=50,
- save_dir=EXP_DIR,
-
- early_stop=False,
-
- use_vdl=True,
-
- resume_checkpoint=None)
|