A method includes receiving training images representing a portion of a drilling rig over a first period of time, associating individual training images of the training images with times at which the individual training images were captured, determining a rig state at each of the times, classifying the individual training images based on the rig state at each of the times, training a machine learning model to identify rig state based on the classified training images, receiving additional images representing the portion of the drilling rig over a second period of time, and determining one or more rig states of the drilling rig during the second period of time using the machine learning model based on the additional images.