Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

UNet with Pascal VOC gives 5% mIOU

See original GitHub issue

Hi, I have been trying to run UNet with Pascal VOC dataset and I am getting really bad results. The same dataset is doing great with PSPNet but UNet results are really bad.

Here is my config:

{
    "name": "UNET",
    "n_gpu": 1,
    "use_synch_bn": true,

    "arch": {
        "type": "UNet",
        "args": {
            "backbone": "resnet50",
            "freeze_bn": false,
            "freeze_backbone": false
        }
    },

    "train_loader": {
        "type": "VOC",
        "args":{
            "data_dir": "/mnt/batch/tasks/shared/LS_root/mounts/clusters/objloc/code/datasets/VOC/",
            "batch_size": 8,
            "base_size": 400,
            "crop_size": 380,
            "augment": true,
            "shuffle": true,
            "scale": true,
            "flip": true,
            "rotate": true,
            "blur": false,
            "split": "train",
            "num_workers": 8
        }
    },

    "val_loader": {
        "type": "VOC",
        "args":{
            "data_dir": "/mnt/batch/tasks/shared/LS_root/mounts/clusters/objloc/code/datasets/VOC/",
            "batch_size": 8,
            "crop_size": 480,
            "val": true,
            "split": "val",
            "num_workers": 4
        }
    },

    "optimizer": {
        "type": "SGD",
        "differential_lr": true,
        "args":{
            "lr": 0.001,
            "weight_decay": 0.0001,
            "momentum": 0.9
        }
    },

    "loss": "CrossEntropyLoss2d",
    "ignore_index": 255,
    "lr_scheduler": {
        "type": "Poly",
        "args": {}
    },

    "trainer": {
        "epochs": 80,
        "save_dir": "/mnt/batch/tasks/shared/LS_root/mounts/clusters/objloc/code/pytorch_segmentation/saved/",
        "save_period": 10,
  
        "monitor": "max Mean_IoU",
        "early_stop": 10,
        
        "tensorboard": true,
        "log_dir": "saved/runs",
        "log_per_iter": 20,

        "val": true,
        "val_per_epochs": 5
    }
}

am I doing something wrong?

Issue Analytics

State:
Created 3 years ago
Reactions:2
Comments:7 (4 by maintainers)

Top GitHub Comments

3reactions

gauridhawancommented, Dec 11, 2020

Hi, we tried using a similar config with learning rate 0.1 and Adam optimizer but it still gives mIoU as 0.3, is there a way to improve it?

0reactions

yassoualicommented, Oct 28, 2020

closed for now, please feel free to open it if you have other questions

Top Results From Across the Web

Semantic Segmentation - Papers With Code

Some example benchmarks for this task are Cityscapes, PASCAL VOC and ADE20K. Models are usually evaluated with the Mean Intersection-Over-Union (Mean IoU) and ......

arXiv:1804.03821v1 [cs.CV] 11 Apr 2018

With ResNeXt 131, we get 0.8% performance gain and achieve 80.8% mIoU when training with 10582 images from PASCAL VOC 2012 [20] and...

keyurparalkar/Semantic-Segmentation-with-UNETs - GitHub

This project focuses on implementing semantic segmentation on pascal VOC dataset using UNET. Implementation of UNET is achieved with fastai library.

A Multi-Attention UNet for Semantic Segmentation in Remote ...

PSPNet is a pyramid scene parsing network for the multiscale problem, and it yielded a new record of mIoU accuracy of 85.4% on...

Co-Occurrent Features in Semantic Segmentation

on Pascal Context, 87.2% mIoU on Pascal VOC 2012 and. 44.89% mIoU on ADE20K datasets. ... The top-1 and top-5 accuracy (%) on...