B ] @sPddlZddlZddlmZmZmZdddZdddZ dd d Z dd d Z dS)N) path_utils midi_utils display_utils cCsjd}t|}tj|tjd}tjjt||dd}||}tj|dddf}t d t|||fS) z%Load the samples used for evaluation.z./dataset/eval.npy)dtypeF)replaceizLoaded {} melody samples) nploadasarrayfloat32randomchoicelentftruncated_normalprintformat)Zn_sampleZsample_source_pathdataZ random_idxZsample_xZsample_zr7/Users/mzliang/Downloads/Lab 2/utils/inference_utils.pyload_melody_sampless rcCs$|dkrtjd}|||fddS)N)rr iF)training)rrr) generatorconditioned_trackZ noise_vectorrrrgenerate_pianoroll+s r*./Experiments/data/happy_birthday_easy.midcCs8tj|d}t||}tj|d}tj||d|S)N)midi) saveto_dir)destination_path)rget_conditioned_trackrrnew_temp_midi_pathZsave_pianoroll_as_midinumpy)rr input_midi_filergenerated_pianorollr!rrr generate_midi1s    r'cCs6tj|d}x$t|D]}t||}t|qWdS)N)r)rr"rangerrshow_pianoroll)reval_dirr%Zn_prrir&rrrshow_generated_pianorolls9s  r-)r)N)r)rr() tensorflowrr$r utilsrrrrrr'r-rrrrs