(mind) xiaoyao@xiaoyao:~/mindyolo-master$ python demo/predict.py --config /home/xiaoyao/mindyolo-master/datasets/mydata.yaml --weight=/home/xiaoyao/mindyolo-master/runs/2024.11.01-10.08.07/weights/mydata-46_366.ckpt --image_path /home/xiaoyao/mindyolo-master/datasets/images/train/00011.jpg --device_target=GPU --iou_thres=0.5 --conf_thres=0.01
2024-11-01 16:09:23,538 [WARNING] Parse Model, args: nearest, keep str type
2024-11-01 16:09:23,548 [WARNING] Parse Model, args: nearest, keep str type
2024-11-01 16:09:23,611 [INFO] number of network params, total: 3.025151M, trainable: 3.014732M
2024-11-01 16:09:26,134 [INFO] Load checkpoint from [/home/xiaoyao/mindyolo-master/runs/2024.11.01-10.08.07/weights/mydata-46_366.ckpt] success.
2024-11-01 16:09:28,166 [INFO] Predict result is: {'category_id': [], 'bbox': [], 'score': []}
2024-11-01 16:09:28,166 [INFO] Speed: 2013.5/2.1/2015.7 ms inference/NMS/total per 640x640 image at batch-size 1;
2024-11-01 16:09:28,166 [INFO] Detect a image success.
2024-11-01 16:09:28,176 [INFO] Infer completed.