How to reproduce DDRNet_23_adb_bag

XuJiacong / PIDNet

This is the official repository for our recent work: PIDNet

MIT License

564 stars 106 forks source link

Open Ruhrozz opened 1 year ago

Ruhrozz commented 1 year ago

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.

zhy168 commented 2 months ago

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