search

HARRIXJANG / SSLFR_master

A self-supervised learning framework for machining feature recognition.
5 stars 0 forks source link

Setup / Dependencies / Requirements #1

Open piw-bot opened 1 week ago

piw-bot commented 1 week ago

The listed setup does not suffice to run the code:

(1) cuda 11.6.112 (2) python 3.8.13 (3) pytorch 1.12.0 (4) tensorboard 2.10.0

Additional packages need to be installed (h5py, matplotlib, scikit-learn, termcolor, timm, chamferdist) but I'm running into two problems when I want to run Predict.py

1) The first line in Loss.py: from chamferdist import ChamferDistance, knn_points results in ImportError: cannot import name 'knn_points' from 'chamferdist'

2) Line 5 from Predict.pyresults in ModuleNotFoundError: No module named 'DataProcessing'

Could you provide a .yml or requirements.txt?

HARRIXJANG commented 6 days ago

Thank you for your question. We have made some corrections to the repository, so please download it again. Specifically, for issues related to chamferdist, you can visit chamferdist to install the corresponding code file, or you can simply use the command pip install chamferdist for installation.

telee6789 commented 1 day ago

image

Try to modify this file.