2 Commits 5bd461392e ... 5f5955c7c1

Author SHA1 Message Date
  zbc 5f5955c7c1 first commit 3 weeks ago
  zbc 5bd461392e first commit 3 weeks ago
1 changed files with 5 additions and 5 deletions
  1. 5 5
      readme.md

+ 5 - 5
readme.md

@@ -12,7 +12,7 @@ This project involves training object detection models using the YOLOv5 framewor
 
 ### 2. Model Training
 - Run the following script to train the model, where `<project_name>` is the name of the newly created project:
-    ```bash
+    ```
     ./object_detection.sh /data2/object_detection/data/image/<project_name>/
     ```
 
@@ -21,22 +21,22 @@ This project involves training object detection models using the YOLOv5 framewor
 
 ### 4. Export Model to `.pb` Format
 - To export the `.pt` model to `.pb` format, first find the `.pt` file path by running:
-    ```bash
+    ```
     cat export.sh
     ```
 - Navigate to the YOLOv5 directory:
-    ```bash
+    ```
     cd /data/object_detection/code/yolov5/
     ```
 - Run the export script to convert the `.pt` model to `.pb` format:
-    ```bash
+    ```
     python export.py --weights ./runs/train/helmet_fall_phone/weights/best.pt --include pb
     ```
   - *Note*: Change `helmet_fall_phone` to the correct project name before running the script.
 
 ### 5. Model Testing
 - To test the model, run the following command:
-    ```bash
+    ```
     python detect.py --weights runs/train/safebelt_data/weights/best.pt --source /data2/object_detection/code/yolov5/safebelt_test --data /data2/object_detection/data/image/safebelt_data/yolo/safebelt_data.yaml --device 0
     ```
   - *Note*: Adjust the `--weights`, `--source`, and `--data` parameters to the correct project paths.