Dimitri Korsch 6 vuotta sitten
vanhempi
commit
561cc5b696
2 muutettua tiedostoa jossa 3 lisäystä ja 2 poistoa
  1. 1 1
      nabirds/__init__.py
  2. 2 1
      scripts/display.py

+ 1 - 1
nabirds/__init__.py

@@ -1,4 +1,4 @@
 from .dataset import Dataset, ImageWrapperDataset
 from .annotations import NAB_Annotations, CUB_Annotations
 
-__version__ = "0.3.1"
+__version__ = "0.3.2"

+ 2 - 1
scripts/display.py

@@ -49,8 +49,9 @@ def main(args):
 
 	start = max(args.start, 0)
 	n_images = min(args.n_images, len(data) - start)
+	idxs = range(start, max(start, start + n_images)
 
-	for i in range(start, max(start, start + n_images)):
+	for i in idxs:
 		im, parts, label = data[i]
 
 		fig1, axs = plt.subplots(2, 1, figsize=(16,9))