For my lab, auto bounding box labeling program.
Using Detectron 2
Detectron provide torch version only 1.8 to 1.10, must be install right torch version. And also must be install right CUDA
Link
Install Detectron2 must be gpu mode. we provide pycocotools for python in this code, but if it is not available, please install the pycocotools
pycocotools
From source | Anaconda | Windows with Conda
Detectron2
pip
pip install -r requirements.txt
Anaconda
conda install -y pyqt jupyter tqdm opencv
Download and Unzip
├── [Detection_Label_Video](./)
│ ├── [weights](./weights)
│ ├── [metrics.json](./weights/metrics.json)
│ ├── [coco_instances_results.json](./weights/coco_instances_results.json)
│ ├── [model_final.pth](./weights/model_final.pth)
│ ├── [model_0002999.pth](./weights/model_0002999.pth)
│ ├── [instances_predictions.pth](./weights/instances_predictions.pth)
│ ├── [last_checkpoint](./weights/last_checkpoint)
│ ├── [events.out.tfevents.1654235188.cvmilab-server.476813.0](./weights/events.out.tfevents.1654235188.cvmilab-server.476813.0])
│ └── [README.md](./weights/README.md)
python main.py
Shortcut key | Name | Do |
---|---|---|
Ctrl + Mouse Left Button | Draw | Drawing the box |
Alt + F4 | Quit | App's Exit |
Ctrl + Enter | Done | App's Done |
s | Start / Pause | If video pause, run or run, pause. |
Ctrl + c | copy | Copy the on Box |
Ctrl + v | paste | Paste the on Box |