HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

ultralytics 8.0.169TQDM, INTERP_LINEAR and RTDETR load_image() updates (#4704)

Browse source 
Co-authored-by: Rustem Galiullin <rustemgal@gmail.com>
Co-authored-by: Rustem Galiullin <rustem.galiullin@bayanat.ai>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Glenn Jocher 2023-09-02 20:01:57 +02:00 • committed by GitHub
parent a4fabfdacf
commit 187b504d68
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
23 changed files with 101 additions and 120 deletions
Download patch file Download diff file Expand all files Collapse all files

7
ultralytics/models/fastsam/prompt.py
View file

@ -8,9 +8,8 @@ import matplotlib.pyplot as plt
import numpy as np
import torch
from PIL import Image
from tqdm import tqdm
from ultralytics.utils import TQDM_BAR_FORMAT
from ultralytics.utils import TQDM
class FastSAMPrompt:
@ -87,7 +86,7 @@ class FastSAMPrompt:
retina=False,
withContours=True):
n = len(annotations)
pbar = tqdm(annotations, total=n, bar_format=TQDM_BAR_FORMAT)
pbar = TQDM(annotations, total=n)
for ann in pbar:
result_name = os.path.basename(ann.path)
image = ann.orig_img
@ -156,7 +155,7 @@ class FastSAMPrompt:
save_path.parent.mkdir(exist_ok=True, parents=True)
cv2.imwrite(str(save_path), img_array)
plt.close()
pbar.set_description('Saving {} to {}'.format(result_name, save_path))
pbar.set_description(f'Saving {result_name} to {save_path}')
@staticmethod
def fast_show_mask(