plt.subplot(4, 3, 12)
plt.imshow(target[:,:,::-1])
plt.axis('off')
fig.savefig(out_path, bbox_inches='tight')
plt.close(fig)
print('out_inds: ', out_inds)
print('out_msks: ', out_msks)
# print('cls_inds: ', entry['cls_inds'])
# print('pos_inds: ', entry['pos_inds'])
# print('trans_inds: ', entry['trans_inds'])
break
if __name__ == '__main__':
cv2.setNumThreads(0)
config, unparsed = get_config()
config = abstract_arguments(config)
np.random.seed(config.seed)
random.seed(config.seed)
torch.manual_seed(config.seed)
if(config.cuda):
torch.cuda.manual_seed_all(config.seed)
prepare_directories(config)
# test_dataset(config)
test_loader(config)
# test_lang_vocab(config)
# test_clip_and_triplet(config)
# test_indices(config)
def setUp(self):
config, unparsed = get_config()
np.random.seed(config.seed)
random.seed(config.seed)
self.cfg = deepcopy(config)