limberc / LightningSoD

FastSoD

This repo has been heavily inherit from this repo. We have modified this repo based on PyTorch Lightning.

The data folder structure is presented as below:

data
├── DUT-OMRON
│   ├── DUT-OMRON-bounding-box
│   ├── DUT-OMRON-eye-fixations
│   ├── images
│   └── masks
└── ECSSD
    ├── images
    └──  masks

Training

python lightning.py fit --model.model {SAMNet OR HVPNet}

Infer

python lightning.py test --model.model {SAMNet OR HVPNet}