iicd/xautodl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update codes

Browse Source
This commit is contained in:
D-X-Y 2021-05-09 19:05:07 +08:00
parent 58dee23a11
commit 870cfc40c2
2 changed files with 21 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
exps/LFNA/vis-synthetic.py
View File

@ -179,11 +179,11 @@ def compare_cl(save_dir):
) )
def visualize_env(save_dir): def visualize_env(save_dir, version):
save_dir = Path(str(save_dir)) save_dir = Path(str(save_dir))
save_dir.mkdir(parents=True, exist_ok=True) save_dir.mkdir(parents=True, exist_ok=True)
dynamic_env = get_synthetic_env() dynamic_env = get_synthetic_env(version=version)
min_t, max_t = dynamic_env.min_timestamp, dynamic_env.max_timestamp min_t, max_t = dynamic_env.min_timestamp, dynamic_env.max_timestamp
for idx, (timestamp, (allx, ally)) in enumerate(tqdm(dynamic_env, ncols=50)): for idx, (timestamp, (allx, ally)) in enumerate(tqdm(dynamic_env, ncols=50)):
dpi, width, height = 30, 1800, 1400 dpi, width, height = 30, 1800, 1400
@ -201,11 +201,15 @@ def visualize_env(save_dir):
tick.label.set_rotation(10) tick.label.set_rotation(10)
for tick in cur_ax.yaxis.get_major_ticks(): for tick in cur_ax.yaxis.get_major_ticks():
tick.label.set_fontsize(LabelSize - font_gap) tick.label.set_fontsize(LabelSize - font_gap)
cur_ax.set_xlim(-10, 10) if version == "v1":
cur_ax.set_ylim(-60, 60) cur_ax.set_xlim(-2, 2)
cur_ax.set_ylim(-60, 60)
elif version == "v2":
cur_ax.set_xlim(-10, 10)
cur_ax.set_ylim(-60, 60)
cur_ax.legend(loc=1, fontsize=LegendFontsize) cur_ax.legend(loc=1, fontsize=LegendFontsize)
save_path = save_dir / "{:05d}".format(idx) save_path = save_dir / "v{:}-{:05d}".format(version, idx)
fig.savefig(str(save_path) + ".pdf", dpi=dpi, bbox_inches="tight", format="pdf") fig.savefig(str(save_path) + ".pdf", dpi=dpi, bbox_inches="tight", format="pdf")
fig.savefig(str(save_path) + ".png", dpi=dpi, bbox_inches="tight", format="png") fig.savefig(str(save_path) + ".png", dpi=dpi, bbox_inches="tight", format="png")
plt.close("all") plt.close("all")
@ -213,8 +217,8 @@ def visualize_env(save_dir):
base_cmd = "ffmpeg -y -i {xdir}/%05d.png -vf scale=1800:1400 -pix_fmt yuv420p -vb 5000k".format( base_cmd = "ffmpeg -y -i {xdir}/%05d.png -vf scale=1800:1400 -pix_fmt yuv420p -vb 5000k".format(
xdir=save_dir xdir=save_dir
) )
os.system("{:} {xdir}/env.mp4".format(base_cmd, xdir=save_dir)) os.system("{:} {xdir}/env-{ver}.mp4".format(base_cmd, xdir=save_dir, ver=version))
os.system("{:} {xdir}/env.webm".format(base_cmd, xdir=save_dir)) os.system("{:} {xdir}/env-{ver}.webm".format(base_cmd, xdir=save_dir, ver=version))
def compare_algs(save_dir, alg_dir="./outputs/lfna-synthetic"): def compare_algs(save_dir, alg_dir="./outputs/lfna-synthetic"):
@ -417,7 +421,8 @@ if __name__ == "__main__":
) )
args = parser.parse_args() args = parser.parse_args()
compare_algs_v2(os.path.join(args.save_dir, "compare-alg-v2")) visualize_env(os.path.join(args.save_dir, "vis-env"), "v1")
# visualize_env(os.path.join(args.save_dir, "vis-env")) visualize_env(os.path.join(args.save_dir, "vis-env"), "v2")
# compare_algs_v2(os.path.join(args.save_dir, "compare-alg-v2"))
# compare_cl(os.path.join(args.save_dir, "compare-cl")) # compare_cl(os.path.join(args.save_dir, "compare-cl"))
# compare_algs(os.path.join(args.save_dir, "compare-alg")) # compare_algs(os.path.join(args.save_dir, "compare-alg"))

14
lib/datasets/synthetic_core.py
View File

@ -29,6 +29,13 @@ def get_synthetic_env(total_timestamp=1000, num_per_task=1000, mode=None, versio
), ),
) )
if version == "v1": if version == "v1":
function = DynamicLinearFunc()
function_param = dict()
function_param[0] = ComposedSinFunc(
amplitude_scale=ConstantFunc(1.0), period_phase_shift=ConstantFunc(10)
)
function_param[1] = ConstantFunc(constant=0.9)
elif version == "v2":
function = DynamicQuadraticFunc() function = DynamicQuadraticFunc()
function_param = dict() function_param = dict()
function_param[0] = ComposedSinFunc( function_param[0] = ComposedSinFunc(
@ -38,13 +45,6 @@ def get_synthetic_env(total_timestamp=1000, num_per_task=1000, mode=None, versio
function_param[2] = ComposedSinFunc( function_param[2] = ComposedSinFunc(
num_sin_phase=5, phase_shift=0.4, max_amplitude=0.9 num_sin_phase=5, phase_shift=0.4, max_amplitude=0.9
) )
elif version == "v2":
function = DynamicLinearFunc()
function_param = dict()
function_param[0] = ComposedSinFunc(
amplitude_scale=ConstantFunc(1.0), period_phase_shift=ConstantFunc(1.0)
)
function_param[1] = ConstantFunc(constant=0.9)
else: else:
raise ValueError("Unknown version: {:}".format(version)) raise ValueError("Unknown version: {:}".format(version))