Adding parameter, to choose amount of upsample layers.

ChristianMarzahl / ObjectDetection

Some experiments with object detection in PyTorch

136 stars 43 forks source link

Adding parameter, to choose amount of upsample layers. #19

Open VCasecnikovs opened 4 years ago

VCasecnikovs commented 4 years ago

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