[update]

.gitignore

@@ -254,3 +254,6 @@ com_crashlytics_export_strings.xml
 crashlytics.properties
 crashlytics-build.properties
 fabric.properties
+
+datasets
+weights

AX620_README.md

@@ -1,5 +1,7 @@
 # ENVIRONMENT INIT
 ```bash
+sudo apt update
+sudo apt install libgl1 libglib2.0-0 git
 pip install -r requirements.txt
 ```
 
@@ -11,9 +13,9 @@ wget -P weights https://github.com/ultralytics/yolov5/releases/download/v6.2/yol
 
 # DOWN DATASETS
 ```bash
-mkdir -p datasets/rubbish
+mkdir -p datasets
 wget -P datasets https://github.com/Abandon-ht/coco_rubbish_dataset/archive/refs/heads/main.zip
-unzip -d datasets/rubbish datasets/coco_rubbish_dataset-main.zip
+unzip -d datasets datasets/main.zip
 ```
 
 # TRAIN
@@ -25,7 +27,7 @@ python train.py --data data/rubbish.yaml --cfg models/yolov5s.yaml --weights wei
 python detect.py --source datasets/rubbish/images/IMG_20210311_213716.jpg --weights ./runs/train/exp/weights/best.pt
 ```
 
-# EXPORT ONNX MODULE
+# EXPORT AX620 ONNX MODULE
 ```bash
 python export.py --include onnx --opset 11 --weights ./runs/train/exp/weights/best.pt ax620
 ```

data/rubbish.yaml

@@ -1,5 +1,5 @@
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: datasets/rubbish  # dataset root dir
+path: datasets/coco_rubbish_dataset-main  # dataset root dir
 train: train2017.txt  # train images (relative to 'path') 118287 images
 val: val2017.txt  # val images (relative to 'path') 5000 images
 # test: test-dev2017.txt  # 20288 of 40670 images, submit to https://competitions.codalab.org/competitions/20794