Пример #1
0
def train(args, model, device, train_loader, test_loader, optimizer, epoch):

    model.train()
    lr = optimizer.param_groups[0]['lr']
    correct = 0
    total = 0
    losses = AverageMeter()

    for i, data in enumerate(train_loader):
        inputs, labels = data[0].to(device), data[1].to(device)
        optimizer.zero_grad()
        outputs = model(inputs)
        loss = nn.CrossEntropyLoss()(outputs, labels)
        loss.backward()
        optimizer.step()
        losses.update(loss.float().mean().item())

        _, predicted = outputs.max(1)
        total += labels.size(0)
        correct += predicted.eq(labels).sum().item()
        acc = correct * 100. / total

        progress_bar(
            i, len(train_loader),
            '[Epoch %d] CE: %.4f | Top 1 Acc: %0.2f%% | LR: %.2e' %
            (epoch, losses.avg, acc, lr))

    return test(args, model, device, test_loader)
Пример #2
0
def test(args, model, device, test_loader):
    model.eval()
    top1 = AverageMeter()
    top5 = AverageMeter()
    with torch.no_grad():
        for i, data in enumerate(test_loader):
            inputs, labels = data[0].to(device), data[1].to(device)
            outputs = model(inputs)

            acc1, acc5 = accuracy(outputs, labels, (1, 5))
            top1.update(acc1[0], inputs.size(0))
            top5.update(acc5[0], inputs.size(0))

            progress_bar(
                i, len(test_loader),
                'Test accuracy Top 1: %0.2f%%, Top 5: %0.2f%%' %
                (top1.avg, top5.avg))
    return top1.avg, top5.avg
Пример #3
0
    def _download(self, url, file_name):
        """
        See :meth:`download`.

        """
        downloads_directory = config.CONFIG["downloads-directory"]
        if utils.prepare_directory(downloads_directory):
            logger.info(
                "Created directory(s) %s.",
                downloads_directory
            )

        # Find an available file path
        final_file_path = utils.find_available_file(
            os.path.join(downloads_directory, file_name)
        )
        final_file_name = os.path.basename(final_file_path)

        logger.debug("File will be saved to %s.", final_file_path)

        # Get a generator function that makes a pretty progress bar.
        bar = ui.progress_bar_indeterminate()

        # Actually try to grab the file from the server
        while True:
            ui.print_carriage(
                "%s Trying to download file... %s" %
                    (ui.progress_bar(0.0), " " * 30)
            )

            # Ask the server for the file
            try:
                file_request = self.requests_session.get(
                    url, timeout = 1, stream = True, verify = _get_verify()
                )
            except requests.exceptions.Timeout:
                logger.info(
                    "Request timed out. Server did not accept connection after "
                    "1 second."
                )

            # If it's giving it to us...
            if file_request.status_code == 200:
                logger.debug(
                    "Response headers...\n%s",
                    pprint.pformat(file_request.headers, width = 72)
                )

                if "content-length" in file_request.headers:
                    size = float(file_request.headers["content-length"])
                else:
                    logger.info("File is of unknown size.")

                    size = 0
                    ui.print_carriage(
                        ui.progress_bar(-1) + " Downloading file."
                    )

                # Download the file in chunks.
                chunk_size = 124
                downloaded = 0
                with open(final_file_path, "wb") as f:
                    for chunk in file_request.iter_content(124):
                        if size != 0:
                            ui.print_carriage(
                                ui.progress_bar(downloaded / size) +
                                " Downloading file."
                            )
                            downloaded += chunk_size

                        f.write(chunk)

            # If the server got particularly angry at us...
            if (file_request.status_code == 500 or
                    ("X-CallSuccess" in file_request.headers and
                    file_request.headers["X-CallSuccess"] == "False")):
                logger.critical(
                    "500 response. The server encountered an error."
                )
                sys.exit(1)

            if file_request.status_code == requests.codes.ok:
                break

            # Make sure that the trying prompt appears for at least a moment or
            # so
            time.sleep(0.5)

            period = 0.1
            wait_for = 4
            for i in xrange(int(wait_for / period)):
                ui.print_carriage(
                    next(bar) + " Download not ready yet. Waiting."
                )

                time.sleep(period)

        print "File saved to %s." % utils.shorten_path(final_file_path)
Пример #4
0
def main():
    config = Null()
    config.keyScrollSpeed = 2000.0
    config.musicVolume = .5

    view = Null()
    view.pos = v((0, 0))
    view.sprites = {}
    view.cursor = {}
    view.units = []
    view.selarea = None
    view.begin_pos = None
    view.scroll_start = None
    view.selection = []
    view.players = []
    view.ct = 0.0
    view.scrollleft = False
    view.scrollright = False
    view.scrollup = False
    view.scrolldown = False
    view.uisprites = []
    view.config = config
    view.tilepalcount = 0
    view.followselection = False

    init(sys.argv, view)

    dt = .001
    mp = 0
    #view.pathfind.set_map(chr(0)*view.mapdim[0]*view.mapdim[1])
    clock = pygame.time.Clock()
    while True:
        for event in pygame.event.get():
            if handle_event(event, view): return

        #view.path = pathf(view.occu)
        view.occu = list([list(x) for x in view.mapw])

        for player in view.players:
            player.food = v(0, 0, 0)
            player.hunger = v(0, 0, 0)
        for unit in view.units:
            unit.update(dt, view)

        if view.scrollleft:
            view.pos = view.pos + v(-1.0, 0.0) * config.keyScrollSpeed * dt
        if view.scrollright:
            view.pos = view.pos + v(1.0, 0.0) * config.keyScrollSpeed * dt
        if view.scrollup:
            view.pos = view.pos + v(0.0, -1.0) * config.keyScrollSpeed * dt
        if view.scrolldown:
            view.pos = view.pos + v(0.0, 1.0) * config.keyScrollSpeed * dt

        if view.followselection and view.selection != []:
            poss = [x.pos for x in view.selection]
            poss = sum(poss) / len(poss)
            view.pos = poss - v(640, 480) / 2

        if view.pos[0] < 0: view.pos[0] = 0
        if view.pos[1] < 0: view.pos[1] = 0
        maxx = view.mapdim[0] * 32 - view.screen.get_width()
        maxy = (view.mapdim[1] - 12) * 32
        if view.pos[0] > maxx: view.pos[0] = maxx
        if view.pos[1] > maxy: view.pos[1] = maxy
        view.pos = v([int(x) for x in view.pos])

        view.screen.fill((0, 0, 0))
        if view.tileset == "platform":
            star.draw(view)
        draw_tiles(view)

        tempunits = []
        for unit in view.units:
            tempunits.append((unit.level, unit.pos[1], unit))
        tempunits.sort()
        view.minimapover = pygame.surface.Surface((128, 128), pygame.SRCALPHA,
                                                  32)
        for unit in tempunits:
            unit[2].draw(view)

        for sprite in view.uisprites:
            sprite.draw(view)
            sprite.update()

        if view.selarea:
            pygame.draw.rect(view.screen, (16, 252, 24), view.selarea, 1)

        #68 w 3 down
        player = view.players[0]
        player.raceindex = 1
        player.minerals = 50
        player.gas = 0
        #print player.food, player.hunger
        icons = [
            (0, player.minerals, (16, 252, 24)),
            (player.raceindex + 1, player.gas, (16, 252, 24)),
        ]
        for raceindex in range(3):
            food = player.food[raceindex]
            hunger = player.hunger[raceindex]
            if not food and not hunger:  # heh
                continue
            if food >= hunger:
                color = (16, 252, 24)
            else:
                color = (200, 24, 24)
            icons.append((raceindex + 4, str(hunger) + "/" + str(food), color))

        sprite = view.sprites["icons"]
        x = 640
        icons.reverse()
        for icon in icons:
            x -= 68
            view.screen.blit(sprite[icon[0]], (x, 3))
            text = font.render(10, str(icon[1]), "game5")
            view.screen.blit(text, (x + 16, 3))

        view.screen.blit(view.sprites['ConsoleTerran'], (0, 0))

        view.screen.blit(view.minimap,
                         v(6, 348) + (64, 64) - v(view.minimap.get_size()) / 2)
        view.screen.blit(view.minimapover,
                         v(6, 348) + (64, 64) - v(view.minimap.get_size()) / 2)
        topleft = v(6, 348) + (64, 64) - v(view.minimap.get_size()) / 2
        pygame.draw.line(view.screen, (204, 204, 208), topleft + (0, -1),
                         topleft + (128, -1))
        bottomright = v(6, 348) + (64, 64) + v(view.minimap.get_size()) / 2
        pygame.draw.line(view.screen, (172, 152, 148), bottomright,
                         bottomright + (-128, 0))

        pygame.draw.rect(
            view.screen, (255, 255, 255),
            (v(6, 348) +
             (64, 64) - v(view.minimap.get_size()) / 2 + view.pos / 32,
             (20, 13)), 1)
        #print view.mapdim

        (173, 389)
        if len(view.selection) == 0:
            pass
        elif len(view.selection) == 1:
            u = view.selection[0]
            if u.id:
                w = view.sprites["Wireframe"][u.id]
                w.set_palette(get_wpalette(u))
                view.screen.blit(w, (168, 388))
            if u.health != None:
                s = str(int(u.health)) + "/" + str(u.maxhealth)
                t = font.render(8, s, "game5", True)
                view.screen.blit(t, v(199, 457) - v(t.get_size()) / 2)
            s = u.name
            t = font.render(10, u.name, "game0", True)
            view.screen.blit(t, v(315, 394) - v(t.get_size()) / 2)
        else:
            pass  #multiple grpwire.grp
        #buttons

        view.screen.blit(view.sprites["TerranButtons"][4], (3, 320))
        view.screen.blit(view.sprites["TerranButtons"][0], (416, 388))

        view.screen.blit(font.render(8, "MENU", "game1"), (435, 394))
        mp += .5
        mp = mp % 10
        view.screen.blit(view.movie[int(mp)], (413, 410))
        #pygame.draw.rect(view.screen, (128,128,128), ((413,410),(60,56)))
        view.screen.blit(view.sprites["TerranOverlay"], (413, 410))

        if debug:
            surf = font.render(8, str(int(clock.get_fps() + .5)), "game0")
            view.screen.blit(surf, (0, 0))

        if pygame.mouse.get_focused():
            view.ct += dt * 8
            cursor = 'CursorArrow'
            u = get_unit_at_point(pygame.mouse.get_pos() + view.pos, view)
            if u:
                if u.player == view.players[0]: cursor = 'CursorTarget2Green'
                elif u.player == -1: cursor = 'CursorTarget2Yellow'
                else: cursor = 'CursorTarget2Red'
            if view.selarea: cursor = 'CursorDrag'
            if view.scroll_start: cursor = None
            if view.sprites['ConsoleTerran'].get_at(
                    pygame.mouse.get_pos()) != (0, 0, 0, 255):
                cursor = "CursorArrow"
            if cursor == "CursorArrow": view.ct = view.ct % 4.0
            elif cursor in ("CursorTarget2Green", "CursorTarget2Yellow",
                            "CursorTarget2Red"):
                view.ct = view.ct % 14.0
            else:
                view.ct = 0

            if cursor:
                sprite = view.sprites[cursor][int(view.ct)]
                view.screen.blit(
                    sprite,
                    v(pygame.mouse.get_pos()) - v(sprite.get_size()) / 2)
        try:
            a
        except NameError:
            a = 0.
        a += dt
        a = a % 5
        view.screen.blit(ui.progress_bar(a / 5, view), (281, 426))
        pygame.display.update()
        pygame.mixer.music.set_volume(view.config.musicVolume)
        #sendp(cPickle.dumps(view.units),view)
        #try:
        # while True:
        #  view.units = cPickle.loads(view.socket.recvfrom(2**20)[0])
        #except socket.error: pass
        if debug:
            clock.tick()
        else:
            clock.tick(30)
        dt = clock.get_time() / 1000.0
Пример #5
0
def main():
    config = Null()
    config.keyScrollSpeed = 2000.0
    config.musicVolume = .5
    
    view = Null()
    view.pos = v((0,0))
    view.sprites = {}
    view.cursor = {}
    view.units = []
    view.selarea = None
    view.begin_pos = None
    view.scroll_start = None
    view.selection = []
    view.players = []
    view.ct = 0.0
    view.scrollleft = False
    view.scrollright = False
    view.scrollup = False
    view.scrolldown = False
    view.uisprites = []
    view.config = config
    view.tilepalcount = 0
    view.followselection = False
    
    init(sys.argv, view)
    
    dt = .001
    mp = 0
    #view.pathfind.set_map(chr(0)*view.mapdim[0]*view.mapdim[1])
    clock = pygame.time.Clock()
    while True:
        for event in pygame.event.get():
            if handle_event(event, view): return
            
            
            
        #view.path = pathf(view.occu)
        view.occu = list([list(x) for x in view.mapw])
        
        for player in view.players:
            player.food = v(0,0,0)
            player.hunger = v(0,0,0)
        for unit in view.units:
            unit.update(dt, view)
            
        if view.scrollleft: view.pos = view.pos + v(-1.0,0.0)*config.keyScrollSpeed*dt
        if view.scrollright: view.pos = view.pos + v(1.0,0.0)*config.keyScrollSpeed*dt
        if view.scrollup: view.pos = view.pos + v(0.0,-1.0)*config.keyScrollSpeed*dt
        if view.scrolldown: view.pos = view.pos + v(0.0,1.0)*config.keyScrollSpeed*dt
        
        if view.followselection and view.selection != []:
            poss = [x.pos for x in view.selection]
            poss = sum(poss)/len(poss)
            view.pos = poss - v(640,480)/2
            
        if view.pos[0] < 0: view.pos[0] = 0
        if view.pos[1] < 0: view.pos[1] = 0
        maxx = view.mapdim[0]*32 - view.screen.get_width()
        maxy = (view.mapdim[1]-12)*32
        if view.pos[0] > maxx: view.pos[0] = maxx
        if view.pos[1] > maxy: view.pos[1] = maxy
        view.pos = v([int(x) for x in view.pos])
        
        view.screen.fill((0,0,0))
        if view.tileset == "platform":
            star.draw(view)
        draw_tiles(view)
        
        tempunits = []
        for unit in view.units:
            tempunits.append((unit.level,unit.pos[1],unit))
        tempunits.sort()
        view.minimapover = pygame.surface.Surface((128,128),pygame.SRCALPHA,32)
        for unit in tempunits:
            unit[2].draw(view)
            
        for sprite in view.uisprites:
            sprite.draw(view)
            sprite.update()
            
        if view.selarea: pygame.draw.rect(view.screen, (16,252,24), view.selarea, 1)
        
        #68 w 3 down
        player = view.players[0]
        player.raceindex = 1
        player.minerals = 50
        player.gas = 0
        #print player.food, player.hunger
        icons = [
            (0, player.minerals, (16, 252, 24)),
            (player.raceindex+1, player.gas, (16, 252, 24)),
        ]
        for raceindex in range(3):
            food = player.food[raceindex]
            hunger = player.hunger[raceindex]
            if not food and not hunger: # heh
                continue
            if food >= hunger:
                color = (16, 252, 24)
            else:
                color = (200,24,24)
            icons.append((raceindex+4, str(hunger)+"/"+str(food), color))
            
        sprite = view.sprites["icons"]
        x = 640
        icons.reverse()
        for icon in icons:
            x -= 68
            view.screen.blit(sprite[icon[0]],(x,3))
            text = font.render(10, str(icon[1]), "game5")
            view.screen.blit(text, (x+16, 3))
            
        view.screen.blit(view.sprites['ConsoleTerran'],(0,0))
        
        view.screen.blit(view.minimap, v(6,348)+(64,64)-v(view.minimap.get_size())/2)
        view.screen.blit(view.minimapover, v(6,348)+(64,64)-v(view.minimap.get_size())/2)
        topleft = v(6,348)+(64,64)-v(view.minimap.get_size())/2
        pygame.draw.line(view.screen, (204,204,208), topleft+(0,-1), topleft+(128,-1))
        bottomright = v(6,348)+(64,64)+v(view.minimap.get_size())/2
        pygame.draw.line(view.screen, (172,152,148), bottomright, bottomright+(-128,0))
        
        pygame.draw.rect(view.screen,(255,255,255), (v(6,348)+(64,64)-v(view.minimap.get_size())/2+view.pos/32,(20,13)),1)
        #print view.mapdim
        
        
        (173,389)
        if len(view.selection) == 0:
            pass
        elif len(view.selection) == 1:
            u = view.selection[0]
            if u.id:
                w = view.sprites["Wireframe"][u.id]
                w.set_palette(get_wpalette(u))
                view.screen.blit(w, (168,388))
            if u.health != None:
                s = str(int(u.health))+"/"+str(u.maxhealth)
                t = font.render(8, s, "game5", True)
                view.screen.blit(t,v(199,457)-v(t.get_size())/2)
            s = u.name
            t = font.render(10, u.name, "game0", True)
            view.screen.blit(t,v(315,394)-v(t.get_size())/2)
        else:
            pass #multiple grpwire.grp
        #buttons
        
        view.screen.blit(view.sprites["TerranButtons"][4], (3,320))
        view.screen.blit(view.sprites["TerranButtons"][0], (416,388))
        
        view.screen.blit(font.render(8,"MENU", "game1"),(435,394))
        mp += .5
        mp = mp % 10
        view.screen.blit(view.movie[int(mp)], (413,410))
        #pygame.draw.rect(view.screen, (128,128,128), ((413,410),(60,56)))
        view.screen.blit(view.sprites["TerranOverlay"], (413, 410))
        
        if debug:
            surf = font.render(8, str(int(clock.get_fps()+.5)), "game0")
            view.screen.blit(surf,(0,0))
            
        if pygame.mouse.get_focused():
            view.ct += dt * 8
            cursor = 'CursorArrow'
            u = get_unit_at_point(pygame.mouse.get_pos()+view.pos,view)
            if u:
                if u.player == view.players[0]: cursor = 'CursorTarget2Green'
                elif u.player == -1: cursor = 'CursorTarget2Yellow'
                else: cursor = 'CursorTarget2Red'
            if view.selarea: cursor = 'CursorDrag'
            if view.scroll_start: cursor = None
            if view.sprites['ConsoleTerran'].get_at(pygame.mouse.get_pos()) != (0,0,0,255):
                cursor = "CursorArrow"
            if cursor == "CursorArrow": view.ct = view.ct % 4.0
            elif cursor in ("CursorTarget2Green","CursorTarget2Yellow","CursorTarget2Red"): view.ct = view.ct % 14.0
            else: view.ct = 0
            
            if cursor:
                sprite = view.sprites[cursor][int(view.ct)]
                view.screen.blit(sprite, v(pygame.mouse.get_pos())-v(sprite.get_size())/2)
        try: a
        except NameError: a = 0.
        a += dt
        a = a % 5
        view.screen.blit(ui.progress_bar(a/5,view),(281,426))
        pygame.display.update()
        pygame.mixer.music.set_volume(view.config.musicVolume)
        #sendp(cPickle.dumps(view.units),view)
        #try:
        # while True:
        #  view.units = cPickle.loads(view.socket.recvfrom(2**20)[0])
        #except socket.error: pass
        if debug:
            clock.tick()
        else:
            clock.tick(30)
        dt = clock.get_time()/1000.0
Пример #6
0
    def _download(self, url, file_name):
        """
        See :meth:`download`.

        """
        downloads_directory = config.CONFIG["downloads-directory"]
        if utils.prepare_directory(downloads_directory):
            logger.info("Created directory(s) %s.", downloads_directory)

        # Find an available file path
        final_file_path = utils.find_available_file(
            os.path.join(downloads_directory, file_name))
        final_file_name = os.path.basename(final_file_path)

        logger.debug("File will be saved to %s.", final_file_path)

        # Get a generator function that makes a pretty progress bar.
        bar = ui.progress_bar_indeterminate()

        # Actually try to grab the file from the server
        while True:
            ui.print_carriage("%s Trying to download file... %s" %
                              (ui.progress_bar(0.0), " " * 30))

            # Ask the server for the file
            try:
                file_request = self.requests_session.get(url,
                                                         timeout=1,
                                                         stream=True,
                                                         verify=_get_verify())
            except requests.exceptions.Timeout:
                logger.info(
                    "Request timed out. Server did not accept connection after "
                    "1 second.")

            # If it's giving it to us...
            if file_request.status_code == 200:
                logger.debug("Response headers...\n%s",
                             pprint.pformat(file_request.headers, width=72))

                if "content-length" in file_request.headers:
                    size = float(file_request.headers["content-length"])
                else:
                    logger.info("File is of unknown size.")

                    size = 0
                    ui.print_carriage(
                        ui.progress_bar(-1) + " Downloading file.")

                # Download the file in chunks.
                chunk_size = 124
                downloaded = 0
                with open(final_file_path, "wb") as f:
                    for chunk in file_request.iter_content(124):
                        if size != 0:
                            ui.print_carriage(
                                ui.progress_bar(downloaded / size) +
                                " Downloading file.")
                            downloaded += chunk_size

                        f.write(chunk)

            # If the server got particularly angry at us...
            if (file_request.status_code == 500
                    or ("X-CallSuccess" in file_request.headers
                        and file_request.headers["X-CallSuccess"] == "False")):
                logger.critical(
                    "500 response. The server encountered an error.")
                sys.exit(1)

            if file_request.status_code == requests.codes.ok:
                break

            # Make sure that the trying prompt appears for at least a moment or
            # so
            time.sleep(0.5)

            period = 0.1
            wait_for = 4
            for i in xrange(int(wait_for / period)):
                ui.print_carriage(
                    next(bar) + " Download not ready yet. Waiting.")

                time.sleep(period)

        print "File saved to %s." % utils.shorten_path(final_file_path)