# 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