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

there are 2 types of training methods

• PytorchLightning
• Native Pytorch

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:)

MIT

This project is used by the following companies:

• You Can Be First One Here :)

• @erfanzar