This commit is contained in:
Glenn Jocher
2019-06-23 22:01:11 +02:00
parent 0f2e136c05
commit 57b616b8b1
3 changed files with 5 additions and 37 deletions
+3 -2
View File
@@ -3,9 +3,10 @@
# New VM
rm -rf yolov3 weights coco
git clone https://github.com/ultralytics/yolov3
git clone https://github.com/cocodataset/cocoapi && cd cocoapi/PythonAPI && make && cd ../.. && cp -r cocoapi/PythonAPI/pycocotools yolov3
git clone https://github.com/NVIDIA/apex && cd apex && pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" . --user && cd .. && rm -rf apex
bash yolov3/weights/download_yolov3_weights.sh && cp -r weights yolov3
bash yolov3/data/get_coco_dataset.sh
git clone https://github.com/cocodataset/cocoapi && cd cocoapi/PythonAPI && make && cd ../.. && cp -r cocoapi/PythonAPI/pycocotools yolov3
sudo shutdown
# Re-clone
@@ -95,7 +96,7 @@ python3 test.py --data ../supermarket2/supermarket2.data --weights ../darknet/ba
# Debug/Development
python3 train.py --data data/coco.data --img-size 320 --single-scale --batch-size 32 --accumulate 2 --epochs 1
python3 train.py --data data/coco.data --img-size 320 --single-scale --batch-size 64 --accumulate 1 --epochs 1 --evolve
python3 test.py --weights weights/latest.pt --cfg cfg/yolov3-spp.cfg --img-size 320
gsutil cp evolve.txt gs://ultralytics
+1 -1
View File
@@ -641,7 +641,7 @@ def plot_targets_txt(): # from utils.utils import *; plot_targets_txt()
x = np.loadtxt('targets.txt', dtype=np.float32)
x = x.T
s = ['x targets','y targets','width targets','height targets']
s = ['x targets', 'y targets', 'width targets', 'height targets']
fig, ax = plt.subplots(2, 2, figsize=(8, 8))
ax = ax.ravel()
for i in range(4):