/torch-counter

PyTorch Object Counter

Primary LanguageJupyter Notebook

PyTorch Counter

This is for a counter utility for counting the number of object (person, car) from a large number of pictures. This counter is written on top of PyTorch.

Installing

Install maskrcnn-benchmark

Install maskrcnn-benchmark(https://github.com/facebookresearch/maskrcnn-benchmark) first following instructions in https://github.com/facebookresearch/maskrcnn-benchmark/blob/master/INSTALL.md

clone torch-counter

conda activate maskrcnn-benchmark
git clone https://github.com/yonghah/torch-counter.git

Getting started

See the notebook

https://github.com/yonghah/torch-counter/counter/demo.ipynb