Should I start learning to use Pytorch?

1

Hello,

I have a project in mind, and to be honest, I am not yet familiar with PyTorch. I am trying to determine if it would be a good solution for this project and whether it’s worth starting to learn it.

I have a set of thousands of “.step” files representing mechanical parts. My goal is to identify and count several 3D features, such as the number of holes in each part. I am wondering if it’s possible to create a model using PyTorch to “train” it, enabling it to identify the number of holes in new parts.

Is this feasible with PyTorch?

Thank you in advance for your insights.

2

For your case, I would not recommend you to learn PyTorch unless you are familiar with machine learning.

Is this feasible with PyTorch?

I would say most likely yes. However, learning PyTorch itself would not give you the ability to create a proper model. PyTorch only helps you to implement your model, but you would possibly need to learn how machine learning works to design a proper model for your goal.

You might want to do some searching to find an open source model or something else instead… Maybe this link: python - Counting Holes in A Mesh - Stack Overflow.