Skip to content

Instantly share code, notes, and snippets.

@davyneven

davyneven/gold-yolo-demo.txt

Created September 28, 2023 11:57
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save davyneven/6a970de7ecf136748f4d17e961234b33 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save davyneven/6a970de7ecf136748f4d17e961234b33 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
gold-yolo-demo.txt
# install gold-YOLO
!git clone https://github.com/huawei-noah/Efficient-Computing.git
%cd Efficient-Computing/Detection/Gold-YOLO
%pip install -qr requirements.txt
# Download COCO-pretrained model
!wget -O Gold_n_dist.pt "https://drive.google.com/uc?export=download&id=1nJ9x8Q1Ic-DnCs0ruBh-Zy3t9-IyO9LR"
# infer on demo images
!sed -i "s/inferer = Inferer(source, weights, device, yaml, img_size, half)/inferer = Inferer(source, False, None, weights, device, yaml, img_size, half)/g" ./tools/infer.py
!python tools/infer.py --weights ./Gold_n_dist.pt --source ./data/images --conf-thres 0.4
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment