Skip to content

Commit 6cf30f3

Browse files
authoredFeb 21, 2023
[Fix] fix dataset_converters scripts int arg passing bug (#1648)
* [Fix] fix dataset_converters scripts bug * [Fix] fix dataset_converters scripts bug
1 parent 9751354 commit 6cf30f3

File tree

3 files changed

+20
-2
lines changed

3 files changed

+20
-2
lines changed
 

‎tools/dataset_converters/df2k_ost/preprocess_df2k_ost_dataset.py

+9-1
Original file line numberDiff line numberDiff line change
@@ -307,23 +307,31 @@ def parse_args():
307307
parser.add_argument('--data-root', help='dataset root')
308308
parser.add_argument(
309309
'--crop-size',
310+
type=int,
310311
nargs='?',
311312
default=480,
312313
help='cropped size for HR images')
313314
parser.add_argument(
314-
'--step', nargs='?', default=240, help='step size for HR images')
315+
'--step',
316+
type=int,
317+
nargs='?',
318+
default=240,
319+
help='step size for HR images')
315320
parser.add_argument(
316321
'--thresh-size',
322+
type=int,
317323
nargs='?',
318324
default=0,
319325
help='threshold size for HR images')
320326
parser.add_argument(
321327
'--compression-level',
328+
type=int,
322329
nargs='?',
323330
default=3,
324331
help='compression level when save png images')
325332
parser.add_argument(
326333
'--n-thread',
334+
type=int,
327335
nargs='?',
328336
default=20,
329337
help='thread number when using multiprocessing')

‎tools/dataset_converters/div2k/preprocess_div2k_dataset.py

+2
Original file line numberDiff line numberDiff line change
@@ -373,11 +373,13 @@ def parse_args():
373373
'--thresh-size',
374374
nargs='?',
375375
default=0,
376+
type=int,
376377
help='threshold size for HR images')
377378
parser.add_argument(
378379
'--compression-level',
379380
nargs='?',
380381
default=3,
382+
type=int,
381383
help='compression level when save png images')
382384
parser.add_argument(
383385
'--n-thread',

‎tools/dataset_converters/reds/crop_sub_images.py

+9-1
Original file line numberDiff line numberDiff line change
@@ -158,23 +158,31 @@ def parse_args():
158158
'--scales', nargs='*', default=[], help='scale factor list')
159159
parser.add_argument(
160160
'--crop-size',
161+
type=int,
161162
nargs='?',
162163
default=480,
163164
help='cropped size for HR images')
164165
parser.add_argument(
165-
'--step', nargs='?', default=240, help='step size for HR images')
166+
'--step',
167+
type=int,
168+
nargs='?',
169+
default=240,
170+
help='step size for HR images')
166171
parser.add_argument(
167172
'--thresh-size',
173+
type=int,
168174
nargs='?',
169175
default=0,
170176
help='threshold size for HR images')
171177
parser.add_argument(
172178
'--compression-level',
179+
type=int,
173180
nargs='?',
174181
default=3,
175182
help='compression level when save png images')
176183
parser.add_argument(
177184
'--n-thread',
185+
type=int,
178186
nargs='?',
179187
default=20,
180188
help='thread number when using multiprocessing')

0 commit comments

Comments
 (0)
Please sign in to comment.