Skip to content

Commit cbbcfbc

Browse files
committed
update
1 parent b2928ca commit cbbcfbc

File tree

5 files changed

+15
-4
lines changed

5 files changed

+15
-4
lines changed

Diff for: MetaSCI/.gitignore

-1
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
# zzh
22
*.ckpt.*
33
checkpoint
4-
dataset/*
54
result/*
65
?bak/
76
_bak/

Diff for: MetaSCI/README.md

+15-3
Original file line numberDiff line numberDiff line change
@@ -9,13 +9,25 @@ This code is for CVPR 2021 paper "MetaSCI: Scalable and Adaptive Reconstruction
99
### Directory
1010

1111
- `dataset\mask` : encoding mask, shape = [M N Cr NUM], i.e., NUM [M N Cr] masks stacked along the last dimension.
12-
13-
12+
- `dataset\orig` : original frames, shape = [M N Cr], i.e., Cr [M N] frames stacked along the last dimension.
1413

1514

1615

1716
### Parameter
1817

1918
- `datadir`: path for training set directory
20-
- `maskpath`: path for encoding mask file
19+
- `maskpath`: path for encoding mask file
20+
21+
22+
23+
### Run
24+
25+
#### Training
26+
27+
1. Prepare the masks: divide the large-scale mask into small patches (w./w.o. overlap) with the same size. Save one of the patch for the base model's training, and save the other patches for the adapting models' training.
28+
2. Train the base model with `main_MetaBaseModel_train.py` and one of the mask patch above (`main_MetaBaseModel_train_parallel.py` can be used for parallel training).
29+
3. Finetune the base model above to get the adapting models with `main_MetaAdaptModel_train.py` and the other mask patches.
30+
31+
#### Test
2132

33+
1. Use `main_MetaBaseModel_test.py` and `main_MetaAdaptModel_test.py ` and corresponding checkpoints and masks to reconstruct the video.

Diff for: MetaSCI/dataset/mask/.gitkeep

Whitespace-only changes.

Diff for: MetaSCI/dataset/orig/.gitkeep

Whitespace-only changes.

Diff for: MetaSCI/dataset/test_set/.gitkeep

Whitespace-only changes.

0 commit comments

Comments
 (0)