- src/
- this folder contains all the .py code files such as
- nets.py (containts classes like VGG(), YOLO(), ResNet() etc)
- train.py (contains classes like VGGTrain(), YOLOTrain() etc)
- test.py (~)
- dataloader.py (contains classes like PASCALVOC(), MNIST() etc)
- this folder contains all the .py code files such as
- demo/
- this folder contains subfolders such as demo1/, demo2/ etc
- inside each folder one can find .ipynb notebook with different experiments
- within each of these jupyter notebook, you can the classes and their functions defined in the src/*.py files
- make sure to include a
__init__.py
empty file in each directory
could also add "! pip install GPUtil" as it is needed package.