search

EscVM / OIDv4_ToolKit

Download and visualize single or multiple classes from the huge Open Images v4 dataset
GNU General Public License v3.0
809 stars 635 forks source link

download() method crashes program for casting NoneType #35

Closed d-j-kendall closed 5 years ago

d-j-kendall commented 5 years ago

Download method has misplaced indent in master

Misplaced indent causes the following

  File "main.py", line 38, in <module>
    bounding_boxes_images(args, DEFAULT_OID_DIR)
  File "/home/dkendall/Documents/senior_design/tools/OIDv4_ToolKit/modules/bounding_boxes.py", line 89, in bounding_boxes_images
    download(args, df_val, folder[i], dataset_dir, class_name, class_code, threads = int(args.n_threads))
TypeError: int() argument must be a string, a bytes-like object or a number, not 'NoneType'

downloading.py:88

                        if not args.n_threads:
                            download(args, df_val, folder[i], dataset_dir, class_name, class_code)
                    else:
                        download(args, df_val, folder[i], dataset_dir, class_name, class_code, threads = int(args.n_threads))'''

Proposed change

                        if not args.n_threads:
                            download(args, df_val, folder[i], dataset_dir, class_name, class_code)
                        else:
                                download(args, df_val, folder[i], dataset_dir, class_name, class_code, threads = int(args.n_threads))