PIDNet
PIDNet copied to clipboard
How to reproduce DDRNet_23_adb_bag
Line 282-283, models/others/ddrnet23_adb_bag.py
#self.bag = model_utils.BagFM(planes * 4, planes * 2, planes * 4)
self.dfm = model_utils.DFM3(planes * 4, planes * 4)
Neither of this classes are available.
just change it : self.bag = model_utils.Bag(planes * 4, planes * 4) self.final_layer = segmenthead(planes * 4, head_planes, num_class) and it will work