smart and strong object detection ai built from scatch with some new features ๐๐ (Using some custom CNNs and NeuralNets with a holy help from darknet and yolo for everysingle part and adding a bit of research to it ๐)
Contributions are always welcome!
See contributing.md
for ways to get started.
Please adhere to this project's code of conduct
.
To train ObjD you don't have to do crazy things you just have to pick one of the models and train that with custom data that you want to train ObjD with :)
Help For Training
ArgParses
cfg
python3 torch_train.py --cfg <config/config.yaml>
data
python3 torch_train.py --data <data/path.yaml>
epochs
python3 torch_train.py --epochs <300>
eval
python3 torch_train.py --eval
debug
python3 torch_train.py --debug
device
python3 torch_train.py --device <'cuda:0'>
auto anchors
python3 torch_train.py --auto-anchors
Model | Param | Accuracy | FPS |
---|---|---|---|
ObjD tiny | ~ M |
accuracy on COCO data 50 % | FPS 80 |
ObjD s | ~ M |
accuracy on COCO data 59 % | FPS 72 |
ObjD n | ~ M |
accuracy on COCO data 68 % | FPS 60 |
ObjD hx | ~ M |
accuracy on COCO data 77 % | FPS 43 |
there are 2 types of training methods
they both have some benefits and the both are hackable to make them more customize and its depends on you to choose which way you want to train your model with but i recommend yo use native pytorch cause that one have better work and stabelity right now
python3 train.py
python3 torch_train.py
Hi there ๐ I like to train deep neural nets on large datasets ๐ง . Among other things in this world:)
This project is used by the following companies:
- You Can Be First One Here :)