- Linux OS (Ubuntu)
- 16GB of RAM
- Nvidia GPU with 4GB or more VRAM
-
For environment setup please follow instructions from here
-
Download the model and config files from here
-
Download pretrained model from here and place in model folder
-
Download pretrained model from here and place in main directory of repository
Download my models from here, then run
python detect.py -i "path to image" --config "path to config" --model "path to model"
For exmaple,
python detect.py -i 1.png --config Mymodels/Model_2/Model_2_config.py --model Mymodels/Model_2/Model_2.pth
Please check the instructions for information
For training the models, I have used code from [here]. To train the models further:
- Cone that repository
- Download cityscapes dataset from here
- Use the configuration files
- Then run
python tools/train.py config_file_path –resume-from model_file_path