search

LiyaoTang / contrastBoundary

Contrastive Boundary Learning for Point Cloud Segmentation (CVPR2022)
MIT License
141 stars 12 forks source link

The version of scikit-learn #35

Open zivlzw opened 1 year ago

zivlzw commented 1 year ago

We built the tensorflow environment of CBL according to the configuration file, but the version of the library was not specified. Can you tell us the version of the scikit-learn, or what model environment the CBL uses? Traceback (most recent call last): File "main.py", line 102, in trainer.train() File "/root/contrastBoundary-master/tensorflow/utils/trainer.py", line 66, in train g = GraphBuilder(config, verbose=self.verbose) File "/root/contrastBoundary-master/tensorflow/utils/tf_graph_builder.py", line 37, in init dataset = self.get_dataset(config, verbose) File "/root/contrastBoundary-master/tensorflow/models/../datasets/tf_s3dis_dataset.py", line 108, in init self.load_subsampled_clouds(config.first_subsampling_dl, verbose=verbose) File "/root/contrastBoundary-master/tensorflow/models/../datasets/tf_s3dis_dataset.py", line 245, in load_subsampled_clouds search_tree = pickle.load(f) AttributeError: Can't get attribute 'EuclideanDistance64' on <module 'sklearn.metrics._dist_metrics' from '/root/miniconda3/envs/cbl/lib/python3.7/site-packages/sklearn/metrics/_dist_metrics.cpython-37m-x86_64-linux-gnu.so'>

LiyaoTang commented 1 year ago

Hi,

Thanks for your interest. Please try with scikit-learn<1 and see if the error persists.

Best, Liyao