Ejemplo n.º 1
0
    def main_loop(self):

        if not internet_on():
            print 'No internet connection.'
            return

        try:
            #input = InputName(self.screen)
            #name = input.get_name()
            name = "02 03 2017"

            self.test_hash = name.lower().replace(' ', '_')
            # self.test_hash = str(random.getrandbits(1))
            self.__generate_logger()

            log_info_game = {
                'game_mode_parameter': self.game_mode,
                'game_helper_parameter': self.helper,
                'entrainment_helper_name': (CHARACTERS[self.helper % 2])[0],
                'player_number': self.test_hash
            }
            logging.info("General=> Starting test - %s", log_info_game)

            train = Train(self.screen)
            train.start()

            practiceGame = PracticeGame(screen=self.screen)
            practiceGame.config(self.test_hash)
            _, rate_avg, energy_avg, f0mean_avg = practiceGame.play()  # RG

            score_screen = InterGameScreen(
                screen=self.screen,
                step=0,
                scores=[practiceGame.human_player.score])
            score_screen.show()

            score_helper_1 = 0
            score_helper_2 = 0

            for j in range(1, CANT_GAMES + 1):
                if modeAdmin == PLAYING_MODE_ONE:
                    game = GameTypeOne(
                        screen=self.screen,
                        rate_base=rate_avg,
                        #  range_base=range_avg,  # RG
                        energy_base=energy_avg,  # RG
                        f0mean_base=f0mean_avg,  # RG
                        entrainment_helper=self.helper,
                        score_h1=score_helper_1,
                        score_h2=score_helper_2)
                elif modeAdmin == PLAYING_MODE_TWO:
                    game = GameTypeTwo(
                        screen=self.screen,
                        rate_base=rate_avg,
                        #  range_base=range_avg,  # RG
                        energy_base=energy_avg,  # RG
                        f0mean_base=f0mean_avg,  # RG
                        entrainment_helper=self.helper,
                        score_h1=score_helper_1,
                        score_h2=score_helper_2)
                game.config(self.test_hash)
                score, rate_avg, energy_avg, f0mean_avg = game.play()  # RG
                self.results.append(score)

                # Recupero los scores de los
                score_helper_1 = game.known_helpers[0].score
                score_helper_2 = game.known_helpers[1].score

                score_screen = InterGameScreen(screen=self.screen,
                                               step=j,
                                               scores=self.results)
                score_screen.show()
        except (KeyboardInterrupt, SystemExit):
            raise
Ejemplo n.º 2
0
from src.Analyze import Analyze
from src.Train import Train
from src.FeatureExtraction import FeatureExtraction

if __name__ == "__main__":
    trainer = Train(datafp="data/np-rnn/")
    trainset = [1, 2, 3, 4, 5, 6, 7, 12, 13, 14, 15, 17, 20]
    validationset = [9, 18, 19, 16, 8, 11]
    testset = [9, 10]
    trainer.trainmodel(trainset, validationset, testset)

    #extractor.load([1,2], "data/np-rnn/")

    # -----------------------------
    # extractor = FeatureExtraction()
    # savedir = "data/np-rnn2"
    # extractor.mkdir(savedir)
    # extractor.save(savedir)

    # -----------------------------
    sounds, sr = Analyze.load("datasets/UrbanSound8K/test/audio/")
    labels = [
        "foreground", "background", "synthesis", "synthesis", "synthesis",
        "synthesis", "synthesis"
    ]
    Analyze.plotwave(labels, sounds)
    Analyze.plotspectrogram(labels, sounds)
    Analyze.plotlogspectrogram(labels, sounds)
    Analyze.plotmelspectrogram(labels, sounds, sr)