fix(dex-ycb): numpy array truth value used in conditional (+1 more)

[andrewwhitecdw]

Small fixes in lib/datasets/dex_ycb.py:

fix: numpy array truth value used in conditional checks

Fix: Replace:

        if cfg.TRAIN.CHROMATIC and cfg.MODE == 'TRAIN' and np.random.rand(1) > 0.1:
            im = chromatic_transform(im)
        if cfg.TRAIN.ADD_NOISE and cfg.MODE == 'TRAIN' and np.random.rand(1) > 0.1:
            im = add_noise(im)

with:

        if cfg.TRAIN.CHROMATIC and cfg.MODE == 'TRAIN' and np.random.rand() > 0.1:
            im = chromatic_transform(im)
        if cfg.TRAIN.ADD_NOISE and cfg.MODE == 'TRAIN' and np.random.rand() > 0.1:
            im = add_noise(im)

fix: numpy array truth value in pixel threshold check

Fix: Replace:

            if len(ind) > 0 and num_pixels[ind] > cfg.TRAIN.MAX_PXIEL_PER_OBJECT:

with:

            if len(ind) > 0 and num_pixels[ind[0]] > cfg.TRAIN.MAX_PXIEL_PER_OBJECT:

Files changed

• lib/datasets/dex_ycb.py