File size: 555 Bytes
d5ba31a 31cab2b d5ba31a 31cab2b 1197f7d d5ba31a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import inspect
import torch.nn as nn
from yolo.model import module
def auto_pad():
raise NotImplementedError
def get_layer_map():
"""
Dynamically generates a dictionary mapping class names to classes,
filtering to include only those that are subclasses of nn.Module,
ensuring they are relevant neural network layers.
"""
layer_map = {}
for name, obj in inspect.getmembers(module, inspect.isclass):
if issubclass(obj, nn.Module) and obj is not nn.Module:
layer_map[name] = obj
return layer_map
|