Object localization

Object localization problem solved using Pre-Activation Resnet architecture. Images are 64x64 in RGB mode. Label rectangles are self-made.

Demo

Run Locally

Clone the project

  git clone https://github.com/MyNameIsArko/object-localization

Go to the project directory

  cd Object-localization

Install dependencies

  pip install ray[air] numpy pytorch torchvision tensorboard pillow shap tqdm

Run the notebook!

Acknowledgements

• Human faces images
• Pytorch documentation
• Ray documentation

License

MIT