CNC-Net_PyTorch

This repository contains the official code to reproduce the results from the paper:

𝐂𝐍𝐂-𝐍𝐞𝐭: 𝐒𝐞𝐥𝐟-𝐒𝐮𝐩𝐞𝐫𝐯𝐢𝐬𝐞𝐝 𝐋𝐞𝐚𝐫𝐧𝐢𝐧𝐠 𝐟𝐨𝐫 𝐂𝐍𝐂 𝐌𝐚𝐜𝐡𝐢𝐧𝐢𝐧𝐠 𝐎𝐩𝐞𝐫𝐚𝐭𝐢𝐨𝐧𝐬

[CVF Open Access][arXiv] [YouTube]

Installation

Clone this repository into any place you want.

git clone https://github.com/myavartanoo/CNC-Net_PyTorch.git
cd CNC-Net_PyTorch

Dependencies

Install the dependencies:

• Python 3.8.18
• PyTorch 1.12.1
• numpy
• open3d
• pytorch3d
• pysdf

Quick Start

To run the model for a given object follow the below command.

python run.py --input_object {objec_name} --gpu {device_index} --experiment {experiment_name}

For example:

python run.py --input_object '0.off' --gpu 0 --experiment exp_0

Citation

If you find our code or paper useful, please consider citing:

@inproceedings{CNC-Net,
  title={CNC-Net: Self-Supervised Learning for CNC Machining Operations},
  author={Mohsen Yavatanoo and Sangmin Hong and Reyhaneh Neshatavar and Kyoung Mu Lee},
  booktitle={The IEEE / CVF Computer Vision and Pattern Recognition Conference (CVPR)},
  year={2024}
}