ObjectDetection
ObjectDetection copied to clipboard
Adding parameter, to choose amount of upsample layers.
Added parameter n_upsample_layers - which is the amount of upsample layers in the network. _get_last_upsample_index - is used to find stop index in iterator in Merges _get_size_change_layers_indexes - is used to find indexes of all layers where the grid size was changed Made some testing on ResNet18 ResNet50 ResNet105 everywhere works