The paper discusses the different shapes of diamond segments for stone cutting machines. The purpose of this research is to find ways to improve the abrasion resistance of the diamond segments, reduce the vibration load and increase the cutting efficiency. The operating conditions of stone-cutting machines are analysed, as well as the ways of increasing the efficiency of diamond segments by changing their shape. A calculation of the loads arising during operation of the diamond segments is made, and the results are verified using an experimental setup. The initial shape is analysed and the stresses that exceed the material parameters are found. In order to reduce the stresses, a diamond segment with rounded edges is simulated. Segments with bevelled edges and conical shape are also considered. Based on the experimental results, it is found that one of the promising shapes is that with rounded edges.