A simple object localisation algorithm that can also be applied to other datasets.
Data augmentation
Toloka WaterMeters dataset, download here
put it in the dataset folder and use IO.py to generate image.txt file
${ROOT}
└── src/
├── ckpt/
│
├── dataset/
│ ├── image.txt
│ ├── data.tsv
│ ├── images/
│ ├── collage/
│ ├── masks/
|
├── generator/
│ ├── data_generator.py
│ ├── IO.py
│ ├── utils.py
│
│
├── model/
│ ├── DNN.py
│ ├── losses.py
|
├── results
├── demo.py
├── param.py
├── test.py
└── train.py
├── README.md
└── requirements.txt
[1] CenterNet: Objects as Points
[2] CenterNet: Keras Implementation