| 1234567891011121314151617181920212223242526272829303132 |
- from models.wirenet.wirepoint_dataset import WirePointDataset
- from models.config.config_tool import read_yaml
- import matplotlib.pyplot as plt
- # image_file = "D:/python/PycharmProjects/data"
- #
- # label_file = "D:/python/PycharmProjects/data/labels/train"
- # dataset_test = WireDataset(image_file)
- # dataset_test.show(0)
- # for i in dataset_test:
- # print(i)
- cfg = 'wirenet.yaml'
- cfg = read_yaml(cfg)
- print(f'cfg:{cfg}')
- print(cfg['model']['n_dyn_negl'])
- # net = WirepointPredictor()
- dataset = WirePointDataset(dataset_path=cfg['io']['datadir'], dataset_type='val')
- # dataset.show(0)
- # for i in range(len(dataset)):
- # # dataset.show(i)
- for i in dataset:
- # dataset.show(i)
- # print(i)
- print(i[1]['wires']['line_map'].shape)
- plt.show(dataset['wires']['line_map'])
|
