def main()

in main/test.py [0:0]

• 17 lines of code
• 4 McCabe index (conditional complexity)

def main():

    args = parse_args()
    cfg.set_args(args.subject, args.gpu_ids)
    cudnn.benchmark = True

    tester = Tester(args.test_epoch)
    tester._make_batch_generator()
    tester._make_model()
    
    with torch.no_grad():
        for itr, (inputs, targets, meta_info) in enumerate(tqdm(tester.batch_generator)):

            # forward
            out = tester.model(inputs, targets, meta_info, 'test')
            
            vis = True
            if vis:
                filename = str(itr)
                for bid in range(len(out['mesh_out_refined'])):
                    img = inputs['img'][bid].detach().cpu().numpy().transpose(1,2,0)[:,:,::-1]*255
                    cv2.imwrite(filename + '.jpg', img)
                    tester.mesh.save_obj(out['mesh_out_refined'][bid].cpu().numpy(), None, filename + '_refined.obj')