SFNO-BVMC / scripts /fake_data.py
zhangrenchao's picture
Publish SFNO-BVMC reproduction
1ded4dd verified
Raw
History Blame Contribute Delete
689 Bytes
from pathlib import Path
import json,sys
ROOT=Path(__file__).resolve().parents[1];sys.path.insert(0,str(ROOT))
from model.sfno_bvmc import *
c=load_config(ROOT);path=ROOT/c["data"]["manifest"];path.parent.mkdir(parents=True,exist_ok=True);path.write_text(json.dumps({"format_version":c["data"]["format_version"],"logical_input_shape":[77,721,1440],"logical_target_shape":[74,721,1440],"time_step_hours":6,"training_years":c["data"]["training_years"],"validation_year":2018,"test_year":2020,"tile_size":c["data"]["tile_size"],"tile_origins":c["data"]["tile_origins"],"is_complete_global":False},indent=2));x=state(c["data"]["tile_origins"][0],32,0,0);assert x.shape==(77,32,32);print(path)