https://github.com/mostoriginalign/jetsonsigndetect
https://github.com/mostoriginalign/jetsonsigndetect
Last synced: about 1 year ago
JSON representation
- Host: GitHub
- URL: https://github.com/mostoriginalign/jetsonsigndetect
- Owner: MostOriginalIGN
- Created: 2022-07-09T18:26:02.000Z (almost 4 years ago)
- Default Branch: main
- Last Pushed: 2022-10-23T18:57:35.000Z (over 3 years ago)
- Last Synced: 2025-02-13T10:46:03.652Z (over 1 year ago)
- Language: Python
- Size: 4.19 MB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# jetsonSignDetect
Was a project for an AI and Machine Learning class.
A prototype program to detect speed limit signs. Images are drawings of speed limit signs so it will not work in the real world. Model has accuracy of 94% for me.
## How to run
### Clone repo
`git clone https://github.com/Astrogamer54/jetsonSignDetect.git`
`cd jetsonSignDetect`
### Unpack images
```
mkdir data/traffic_signs/TEST && tar -xvzf data/traffic_signs/TEST.tar.gz -C data/traffic_signs/TEST
tar -xvzf traffic_signs.tar.gz -C data/traffic_signs/
```
### Run train.py
```
python3 train.py -b 8 --epochs CHANGETHISTOHOWMANYEPOCHESYOUWANT -j 2
```
### set env
```
NET=models/traffic_signs
DATASET=data/traffic_signs
```
### Run imageNet
`python3 getsign.py --model=$NET/resnet18.onnx --input_blob=input_0 --output_blob=output_0 --labels=$DATASET/labels.txt $DATASET/TEST/60/200.jpg.jpg sign.png`
### batchresize.py is for resizing images if you take new ones
ex
`python3 batchresize.py DIR`
`python3 batchresize.py data/traffic_signs/40mph`