Args: dataloader (Dataloader): evaluate dataloader. img_size (int): image size after preprocess. images are resized to squares whose shape is (img_size, img_size). confthre (float): confidence threshold ranging from 0 to 1, which
(
self, args, dataloader, img_size, confthre, nmsthre, num_classes)
| 59 | """ |
| 60 | |
| 61 | def __init__( |
| 62 | self, args, dataloader, img_size, confthre, nmsthre, num_classes): |
| 63 | """ |
| 64 | Args: |
| 65 | dataloader (Dataloader): evaluate dataloader. |
| 66 | img_size (int): image size after preprocess. images are resized |
| 67 | to squares whose shape is (img_size, img_size). |
| 68 | confthre (float): confidence threshold ranging from 0 to 1, which |
| 69 | is defined in the config file. |
| 70 | nmsthre (float): IoU threshold of non-max supression ranging from 0 to 1. |
| 71 | """ |
| 72 | self.dataloader = dataloader |
| 73 | self.img_size = img_size |
| 74 | self.confthre = confthre |
| 75 | self.nmsthre = nmsthre |
| 76 | self.num_classes = num_classes |
| 77 | self.args = args |
| 78 | |
| 79 | def evaluate( |
| 80 | self, |
nothing calls this directly
no outgoing calls
no test coverage detected