first commit

train.sh

@@ -0,0 +1,9 @@
+export CUDA_VISIBLE_DEVICES=5,6
+python /data6/zhangw/yolov5_birdproject/train.py \
+  --img 960 \
+  --batch 16 \
+  --epochs 100 \
+  --data /data6/zhangw/yolov5_birdproject/data/coco128.yaml \
+  --weights /data6/zhangw/yolov5_birdproject/yolov5s6.pt \
+  --name KTY_yolov5s6_fromYNDC_addKTYneg_e100 \
+  --device 5,6