dyd1234

🎯

Focusing

dyd1234 / CSPDarknet-53.py

Created July 16, 2024 11:28

Darknet -53 and CSPDarknet-53 SCN with different implementations

	import math
	from collections import OrderedDict

	import torch
	import torch.nn as nn
	import torch.nn.functional as F

	from torchviz import make_dot

	#-------------------------------------------------#

dyd1234 / multi_repair.py

Created November 5, 2024 02:34

Repair for the rotation and scaling

	# --------------------------------------------------------------------
	# ... some modules and imports is not showing here
	# --------------------------------------------------------------------

	# REPAIR implementations
	def is_batchnorm(module): #
	return isinstance(module, nn.modules.batchnorm._BatchNorm)

	# if we need to do with rotation?
	# basic implementation