Spaces:

Johnnyyyyy56
/

ReLeM-FoodSeg103-Demo

Running

ReLeM-FoodSeg103-Demo / tests /test_model.py

Initial clean push of functional ReLeM code and weights

ca8a9df 2 months ago

142 Bytes

	import torch

	pthfile = r'/SEG/mmsegmentation/pretrained_model/recipe1MP_R50.pth'


	net=torch.load(pthfile)

	for k in net.keys():
	print(k)