search

clovaai / ext_portrait_segmentation

MIT License
239 stars 68 forks source link

visualizing results #18

Open ucb-pb opened 4 years ago

ucb-pb commented 4 years ago

Setting --visualize to True doesn't appear to do anything.

HYOJINPARK commented 4 years ago

Setting --visualize to True doesn't appear to do anything.

Do you mean that the visdom not working? when you run : python -m visdom.server ???

ucb-pb commented 4 years ago

Sorry, I mean for running Test_model.py

HYOJINPARK commented 4 years ago

Oh yes I did not add the visualization code in Test_model.py

You can edit by referring the main.py like this (line 487)

if args.visualize:
            if train_config["loss"] == "Lovasz":
                grid_outputs = torchvision.utils.make_grid(color_transform((save_est[0] > 0).cpu().data), nrow=6)
            else:
                grid_outputs = torchvision.utils.make_grid(color_transform(save_est[0].unsqueeze(0).cpu().max(1)[1].data), nrow=6)

            my_logger.image_summary(torchvision.utils.make_grid(save_input[0], normalize=True),
                             opts = dict(title=f'VAL img (epoch: {epoch})',caption=f'VAL img (epoch: {epoch})'))
            my_logger.image_summary(grid_outputs,
                                      opts=dict(title=f'VAL output (epoch: {epoch}, step: {str(mIOU_val)})',
                                                caption=f'VAL output (epoch: {epoch}, step: {str(mIOU_val)})', ))

            grid_gt = torchvision.utils.make_grid((100 * save_gt[0].cpu()).type('torch.ByteTensor').data,
                                                  nrow=6)
            my_logger.image_summary(grid_gt,
                                      opts=dict(title=f'VAL gt (epoch: {epoch}, step: {str(mIOU_val)})',
                                                caption=f'VAL gt (epoch: {epoch}, step: {str(mIOU_val)})', ))
HYOJINPARK commented 4 years ago

Hello @NhatRio

save_est is the output of the validation function. save_est is an estimation result from model.

Thanks for your asking

On Wed, 19 Aug 2020 at 16:25, NhatRio notifications@github.com wrote:

Hi @HYOJINPARK https://github.com/HYOJINPARK, Thanks a lot for your impressive work. What is "save_est" here? I tried your code above but there is a error from "save_est". Have a nice day!

— You are receiving this because you were mentioned. Reply to this email directly, view it on GitHub https://github.com/clovaai/ext_portrait_segmentation/issues/18#issuecomment-675901556, or unsubscribe https://github.com/notifications/unsubscribe-auth/AF5NIP3GLOX5GOTF2XLREC3SBN46PANCNFSM4MCOQERQ .