This is the baseline code for C4MTS Task1.
- Switch to GPU: i. Click on the “Runtime” menu at the top of the screen. ii. Select “Change runtime type” from the drop down menu. iii. In the dialog box that appears, select “GPU” as the hardware accelerator. iv. Click on the “Save” button.
- Yolo_to_pixel_function is just used for visualizing the bounding boxes.
- Specify the paths for the training and validation datasets, the number of classes, and the class labels in the data.yaml file.