新增环境相关的脚本

USAGE.md

@@ -0,0 +1,11 @@
+
+
+```bash
+conda create -n yolov5 python=3.10 -y
+```
+
+```bash
+python3 -m pip install --upgrade pip
+pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu121
+pip3 install -r requirements.txt
+```

convert-voc-yolov5.py

@@ -19,7 +19,7 @@ def get_args():
     args.add_argument(
         "--raw_data_path",
         type=str,
-        default="resource/shipdata",
+        default="resource/ships",
         help="raw data path",
     )
     return args.parse_args()

export-onnx.py

@@ -0,0 +1,10 @@
+import onnxruntime as ort
+import numpy as np
+
+ort_session = ort.InferenceSession("alexnet.onnx")
+
+outputs = ort_session.run(
+    None,
+    {"actual_input_1": np.random.randn(10, 3, 224, 224).astype(np.float32)},
+)
+print(outputs[0])

requirements.txt

@@ -12,8 +12,10 @@ PyYAML>=5.3.1
 requests>=2.23.0
 scipy>=1.4.1
 thop>=0.1.1  # FLOPs computation
-torch>=1.8.0  # see https://pytorch.org/get-started/locally (recommended)
-torchvision>=0.9.0
+# torch>=1.8.0  # see https://pytorch.org/get-started/locally (recommended)
+# torchvision>=0.9.0
+torch --index-url https://download.pytorch.org/whl/cu118
+torchvision --index-url https://download.pytorch.org/whl/cu118
 tqdm>=4.64.0
 ultralytics>=8.0.147
 # protobuf<=3.20.1  # https://github.com/ultralytics/yolov5/issues/8012
@@ -47,3 +49,5 @@ setuptools>=65.5.1 # Snyk vulnerability fix
 # mss  # screenshots
 # albumentations>=1.0.3
 # pycocotools>=2.0.6  # COCO mAP
+
+

train.py

@@ -489,7 +489,7 @@ def main(opt, callbacks=Callbacks()):
     # Checks
     if RANK in {-1, 0}:
         print_args(vars(opt))
-        check_git_status()
+        # check_git_status()
         check_requirements(ROOT / 'requirements.txt')
 
     # Resume (from specified or most recent last.pt)

train.sh

@@ -1,6 +1,6 @@
 python3 train.py \
     --cfg models/yolov5s.yaml --weights weights/yolov5s.pt \
-    --data resource/shipdata-yolov5/data.yaml \
-    --batch-size 16 --epochs 100 \
+    --data resource/ships-yolov5/data.yaml \
+    --batch-size 32 --epochs 30 \
     --device 0
 

transmit.sh

@@ -2,4 +2,4 @@
 scp -P $IP_N4090_SERVER_PORT -r zhr@$IP_ANTIS_PUBLIC_IP:~/project/yolov5 .
 
 # send to server
-scp -P $IP_N3080_SERVER_PORT -r ../yolov5 zhr@$IP_ANTIS_PUBLIC_IP:~/project/yolov5
+scp -P $IP_N3080_SERVER_PORT -r ../yolov5 zhr@$IP_ANTIS_PUBLIC_IP:~/project/yolov5/

utils/general.py

@@ -542,7 +542,7 @@ def check_dataset(data, autodownload=True):
             dt = f'({round(time.time() - t, 1)}s)'
             s = f"success ✅ {dt}, saved to {colorstr('bold', DATASETS_DIR)}" if r in (0, None) else f'failure {dt} ❌'
             LOGGER.info(f'Dataset download {s}')
-    check_font('Arial.ttf' if is_ascii(data['names']) else 'Arial.Unicode.ttf', progress=True)  # download fonts
+    # check_font('Arial.ttf' if is_ascii(data['names']) else 'Arial.Unicode.ttf', progress=True)  # TODO: download fonts
     return data  # dictionary