├── PSOL_inference.py ├── PSOL_training.py ├── README.md ├── generate_box_imagenet.py ├── loader ├── ddt_imagenet_dataset.py └── imagenet_loader.py ├── models └── models.py ├── poster.png ├── requirements.txt └── utils ├── IoU.py ├── augment.py ├── func.py ├── nms.py └── vis.py /PSOL_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tzzcl/PSOL/HEAD/PSOL_inference.py -------------------------------------------------------------------------------- /PSOL_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tzzcl/PSOL/HEAD/PSOL_training.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tzzcl/PSOL/HEAD/README.md -------------------------------------------------------------------------------- /generate_box_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tzzcl/PSOL/HEAD/generate_box_imagenet.py -------------------------------------------------------------------------------- /loader/ddt_imagenet_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tzzcl/PSOL/HEAD/loader/ddt_imagenet_dataset.py -------------------------------------------------------------------------------- /loader/imagenet_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tzzcl/PSOL/HEAD/loader/imagenet_loader.py -------------------------------------------------------------------------------- /models/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tzzcl/PSOL/HEAD/models/models.py -------------------------------------------------------------------------------- /poster.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tzzcl/PSOL/HEAD/poster.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | pillow 2 | numpy 3 | matplotlib 4 | sklearn -------------------------------------------------------------------------------- /utils/IoU.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tzzcl/PSOL/HEAD/utils/IoU.py -------------------------------------------------------------------------------- /utils/augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tzzcl/PSOL/HEAD/utils/augment.py -------------------------------------------------------------------------------- /utils/func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tzzcl/PSOL/HEAD/utils/func.py -------------------------------------------------------------------------------- /utils/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tzzcl/PSOL/HEAD/utils/nms.py -------------------------------------------------------------------------------- /utils/vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tzzcl/PSOL/HEAD/utils/vis.py --------------------------------------------------------------------------------