.. | ||
models | ||
new_bbox | ||
plots | ||
results | ||
.gitignore | ||
dataloader.py | ||
environment.yml | ||
outfile | ||
README.md | ||
test.py | ||
train.py | ||
utils.py |
BOSS
Installing Dependencies
Run conda env create -f environment.yml
.
New bounding box annotations
We re-extracted bounding box annotations using Yolo-v8. The new data is in new_bbox/
. The rest of the data can be found here.
Train
python train.py
. You can specify different arguments in the parser.
Test
python test.py
. You can specify different arguments in the parser.
Resources
The original project page for the BOSS dataset can be found here. Our code is based on the original implementation.