Python-MuseGAN用于乐曲生成的AI

MuseGAN - 用于乐曲生成的AI

“““This script performs inference from a trained model.“““
import os
import logging
import argparse
from pprint import pformat
import numpy as np
import scipy.stats
import tensorflow as tf
from musegan.config import LOGLEVEL LOG_FORMAT
from musegan.data import load_data get_samples
from musegan.model import Model
from musegan.utils import make_sure_path_exists load_yaml update_not_none
LOGGER = logging.getLogger（“musegan.inference“）

def parse_arguments（）:
    “““Parse and return the command line arguments.“““
    parser = argparse.ArgumentParser（）
    parser.add_argument（‘--result_dir‘
                        help=“Directory where the results are saved.“）
    parser.add_argument（‘--checkpoint_dir‘
                        help=“Directory that contains checkpoints.“）
    parser.add_argument（‘--params‘ ‘--params_file‘ ‘--params_file_path‘
                        help=“Path to the file that defines the “
                             “hyperparameters.“）
    parser.add_argument（‘--config‘ help=“Path to the configuration file.“）
    parser.add_argument（‘--runs‘ type=int default=“1“
                        help=“Times to run the inference process.“）
    parser.add_argument（‘--rows‘ type=int default=5
                        help=“Number of images per row to be generated.“）
    parser.add_argument（‘--columns‘ type=int default=5
                        help=“Number of images per column to be generated.“）
    parser.add_argument（‘--lower‘ type=float default=-2
                        help=“Lower bound of the truncated normal random “
                             “variables.“）
    parser.add_argument（‘--upper‘ type=float default=2
                        help=“Upper bound of the truncated normal random “
                             “variables.“）
    parser.add_argument（‘--gpu‘ ‘--gpu_device_num‘ type=str default=“0“
                        help=“The GPU device number to use.“）
    args = parser.parse_args（）
    return args

def setup（）:
    “““Parse command line arguments load model parameters load configurations
    and setup environment.“““
    # Parse the command line arguments
    args = parse_arguments（）

    # Load parameters
    params = load_yaml（args.params）

    # Load training configurations
    config = load_yaml（args.config）
    update_not_none（config vars（args））

    # Set unspecified schedule steps to default values
    for target in （config[‘learning_rate_schedule‘] config[‘slope_schedule‘]）:
        if target[‘start‘] is None:
            target[‘start‘] = 0
        if target[‘end‘] is None:
            target[‘end‘] = config[‘steps‘]

    # Make sure result directory exists
    make_sure_path_exists（config[‘result_dir‘]）

    # Setup GPUs
    os.environ[“CUDA_VISIBLE_DEVICES“] = config[‘gpu‘]

    return params config

def main（）:
    “““Main function.“““
    # Setup
    logging.basicConfig（level=LOGLEVEL format=LOG_FORMAT）
    params config = setup（）
    LOGGER.info（“Using parameters:\n%s“ pformat（par

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     目录           0  2019-07-30 10:11  musegan-master\
     文件          62  2019-07-30 10:11  musegan-master\.gitattributes
     文件        1228  2019-07-30 10:11  musegan-master\.gitignore
     文件        1086  2019-07-30 10:11  musegan-master\LICENSE.txt
     文件         553  2019-07-30 10:11  musegan-master\Pipfile
     文件       18914  2019-07-30 10:11  musegan-master\Pipfile.lock
     文件        6209  2019-07-30 10:11  musegan-master\README.md
     目录           0  2019-07-30 10:11  musegan-master\docs\
     文件          83  2019-07-30 10:11  musegan-master\docs\_config.yml
     目录           0  2019-07-30 10:11  musegan-master\docs\_includes\
     文件         151  2019-07-30 10:11  musegan-master\docs\_includes\audio_player.html
     文件         168  2019-07-30 10:11  musegan-master\docs\_includes\icon_link.html
     文件         206  2019-07-30 10:11  musegan-master\docs\_includes\video_player.html
     目录           0  2019-07-30 10:11  musegan-master\docs\_layouts\
     文件        3515  2019-07-30 10:11  musegan-master\docs\_layouts\default.html
     目录           0  2019-07-30 10:11  musegan-master\docs\_sass\
     文件        5837  2019-07-30 10:11  musegan-master\docs\_sass\jekyll-theme-minimal.scss
     文件        1436  2019-07-30 10:11  musegan-master\docs\abstract.md
     目录           0  2019-07-30 10:11  musegan-master\docs\audio\
     文件     7802032  2019-07-30 10:11  musegan-master\docs\audio\best_samples.mp3
     文件     4321581  2019-07-30 10:11  musegan-master\docs\audio\from_scratch_composer.mp3
     文件     4321581  2019-07-30 10:11  musegan-master\docs\audio\from_scratch_hybrid.mp3
     文件     4321581  2019-07-30 10:11  musegan-master\docs\audio\from_scratch_jamming.mp3
     文件     3241485  2019-07-30 10:11  musegan-master\docs\audio\track_conditional_composer.mp3
     文件     3241485  2019-07-30 10:11  musegan-master\docs\audio\track_conditional_hybrid.mp3
     文件     3241485  2019-07-30 10:11  musegan-master\docs\audio\track_conditional_jamming.mp3
     文件        1744  2019-07-30 10:11  musegan-master\docs\data.md
     文件         511  2019-07-30 10:11  musegan-master\docs\dataset.html
     目录           0  2019-07-30 10:11  musegan-master\docs\figs\
     文件      272837  2019-07-30 10:11  musegan-master\docs\figs\evolution.png
     文件      350406  2019-07-30 10:11  musegan-master\docs\figs\hybrid.png
............此处省略464个文件信息