|
@@ -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))
|