瀏覽代碼

new version

Dimitri Korsch 6 年之前
父節點
當前提交
561cc5b696
共有 2 個文件被更改,包括 3 次插入2 次删除
  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 .dataset import Dataset, ImageWrapperDataset
 from .annotations import NAB_Annotations, CUB_Annotations
 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)
 	start = max(args.start, 0)
 	n_images = min(args.n_images, len(data) - start)
 	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]
 		im, parts, label = data[i]
 
 
 		fig1, axs = plt.subplots(2, 1, figsize=(16,9))
 		fig1, axs = plt.subplots(2, 1, figsize=(16,9))